fix(ci): Proper GPU forwarding in docker compose

matter-labs · Jan 14, 2025 · acafefa · acafefa
1 parent 770cad8
commit acafefa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docker-compose-gpu-runner-cuda-12-0.yml b/docker-compose-gpu-runner-cuda-12-0.yml
@@ -59,7 +59,9 @@ services:
       resources:
         reservations:
           devices:
-            - capabilities: [ gpu ]
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 
   postgres:
     image: "postgres:14"

diff --git a/docker-compose-gpu-runner.yml b/docker-compose-gpu-runner.yml
@@ -49,7 +49,9 @@ services:
       resources:
         reservations:
           devices:
-            - capabilities: [ gpu ]
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
   postgres:
     image: "postgres:14"
     command: postgres -c 'max_connections=200'