# version: '3.7'
# services:
#   app:
#     build:
#       context: .
#       args:
#         - work_dir=/app
#     environment:
#       - FLASK_APP_PATH=/app/server.py
#       - FLASK_PORT=5000
#     working_dir: /app
#     volumes:
#       - './:/app'
#     ports:
#       - 5000:5000

services:
  test:
    image: nvidia/cuda:10.2-base
    command: nvidia-smi
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]