use only GPU 0 for now

ErlerPhilipp · Oct 8, 2024 · 603b48e · 603b48e
1 parent cad7df9
commit 603b48e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -87,7 +87,7 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              count: all
+              device_ids: ['0']
               capabilities: [gpu]
     networks:
       - default