1

2026-03-17 12:29:50 +08:00
parent 13903ddd09
commit 7ad093caf7
1 changed files with 6 additions and 4 deletions
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -20,10 +20,12 @@ services:
        reservations:
          devices:
            - driver: nvidia
-              capabilities: [ gpu, compute, video ]
-              device_ids: [ '1' ]
+              device_ids: [ '0' ]
+              capabilities: [ gpu, compute, utility ]  # 移除 video（非必需，减少兼容问题）
    command: >
      bash -c "
        /opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' &&
-        /opt/conda/envs/trellis/bin/python server.py 
-      "
+        /opt/conda/envs/trellis/bin/python server.py
+      "
+#        /opt/conda/envs/trellis/bin/python server.py
+#        tail -f /dev/null