diff --git a/docker-compose.yaml b/docker-compose.yaml
index f975b9e..6f5a5d1 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -20,8 +20,8 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              count: all  # 使用所有可用GPU，也可指定数量如count: 1
               capabilities: [ gpu, compute, video ]
+              device_ids: [ '1' ]
     command: >
       bash -c "
         /opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' &&