增加docker gpu 配置

deploy/docker_compose/docker-compose.yaml

@@ -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:
+  test:
-    build:
+    image: nvidia/cuda:10.2-base
-      context: .
+    command: nvidia-smi
-      args:
+    deploy:
-        - work_dir=/app
+      resources:
-    environment:
+        reservations:
-      - FLASK_APP_PATH=/app/server.py
+          devices:
-      - FLASK_PORT=5000
+            - driver: nvidia
-    working_dir: /app
+              count: 1
-    volumes:
+              capabilities: [gpu]
-      - './:/app'
-    ports:
-      - 5000:5000