Added extra gpu settings to compose file.

CogStack · Apr 17, 2024 · 9c7527a · 9c7527a
1 parent 47823b6
commit 9c7527a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
@@ -27,6 +27,15 @@ services:
       - "5555:5000"
     networks:
       - cognet
+
+    # uncomment this only when you have gpu access  
+    #deploy:
+    #  resources:
+    #    reservations:
+    #      devices:
+    #        - driver: nvidia
+    #          device_ids: ['0']
+    #          capabilities: ["gpu", "utility", "compute", "video"]
 
 networks:
   cognet: