diff --git a/docker-compose.yml b/docker-compose.yml
index 32f4c50..f3b7ae5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,6 +13,6 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              count: all
+              device_ids: [ '1' ]
               capabilities:
                 - gpu
\ No newline at end of file