From 735747032acf64658584163b55bb3e051df3a48c Mon Sep 17 00:00:00 2001
From: zcr <cgzhou@aidlab.hk>
Date: Tue, 17 Mar 2026 12:15:01 +0800
Subject: [PATCH] 1

---
 docker-compose.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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())' &&