docker access to host gpu

catalystneuro · Oct 10, 2023 · d04ac27 · d04ac27
1 parent a274afa
commit d04ac27
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/docker-compose-dev.yml b/docker-compose-dev.yml
@@ -59,9 +59,13 @@ services:
       - ./containers:/app
       - ./results:/results
       - ./logs:/logs
-    runtime: nvidia
-    devices:
-      - "/dev/nvidia0:/dev/nvidia0"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
 
   database:
     image: postgres:latest

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -43,8 +43,13 @@ services:
     volumes:
       - ./results:/results
       - ./logs:/logs
-    devices:
-      - "/dev/nvidia0:/dev/nvidia0"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
 
   database:
     image: postgres:latest