# 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]