Merge pull request #69 from tengerye/patch-1

Author: pkarw

docker-compose.gpu.yml

@@ -87,3 +87,14 @@ services:
       timeout: 10s
       retries: 3
 
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility