1
This commit is contained in:
11
docker-compose.yaml
Normal file → Executable file
11
docker-compose.yaml
Normal file → Executable file
@@ -1,7 +1,7 @@
|
||||
services:
|
||||
trellis:
|
||||
image: zhouchengrong/fida-trellis:latest # 你 commit 的镜像
|
||||
container_name: trellis-dev
|
||||
image: zhouchengrong/fida-trellis-a6000:latest # 你 commit 的镜像
|
||||
container_name: FiDA_3D_Trellis
|
||||
restart: unless-stopped
|
||||
environment:
|
||||
- NVIDIA_VISIBLE_DEVICES=all
|
||||
@@ -24,8 +24,5 @@ services:
|
||||
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
|
||||
# tail -f /dev/null
|
||||
conda run --no-capture-output -n trellis python server.py
|
||||
"
|
||||
Reference in New Issue
Block a user