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