aidlab/FiDA-3D-Trellis
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
zcr
2026-03-17 12:29:50 +08:00
parent 13903ddd09
commit 7ad093caf7
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docker-compose.yaml
View File

@@ -20,10 +20,12 @@ services:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
capabilities: [ gpu, compute, video ] device_ids: [ '0' ]
device_ids: [ '1' ] capabilities: [ gpu, compute, utility ] # 移除 video（非必需，减少兼容问题）
command: > command: >
bash -c " bash -c "
/opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' && /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