diff --git a/docker-compose.yml b/docker-compose.yml
index dfd6bc6..093b01f 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,9 +11,8 @@ x-base_service: &base_service
       resources:
         reservations:
           devices:
-            - driver: nvidia
-              count: all
-              capabilities: [gpu]
+          - capabilities:
+            - gpu
 volumes:
   data:
   output: