diff --git a/docker-compose.yaml b/docker-compose.yaml
index 01f0f7c45cee81accceaf3579c60e6f0deb7e9ed..0e6308a2b2148545ae916f42394ce515fea86563 100755
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -138,7 +138,9 @@ services:
       resources:
         reservations:
           devices:
-            - capabilities: [gpu]
+            - driver: nvidia
+              device_ids: ['1']  
+              capabilities: [gpu]
           cpus: '10' 
           memory: 30G
     ipc: host