@@ -1,15 +1,27 @@
-version: '3.7'
+# 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:
- 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
+ test:
+ image: nvidia/cuda:10.2-base
+ command: nvidia-smi
+ deploy:
+ resources:
+ reservations:
+ devices:
+ - driver: nvidia
+ count: 1
+ capabilities: [gpu]