diff --git a/docker-compose.yaml b/docker-compose.yaml
index 9a65815..0a3f048 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -20,7 +20,7 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              device_ids: [ '0' ]
+              device_ids: [ '1' ]
               capabilities: [ gpu, compute, utility ]  # 移除 video（非必需，减少兼容问题）
     command: >
       bash -c "