diff --git a/docker-compose.yaml b/docker-compose.yaml
index 193add27..cacfb19b 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -46,8 +46,8 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              count: 1
-              capabilities: [ gpu ]
+              capabilities: [ "gpu" ]
+              count: all
 
   db:
     image: postgres:15-alpine