Files
FiDA-3D-Trellis/docker-compose.yaml

28 lines
902 B
YAML
Raw Permalink Normal View History

2026-03-17 11:29:17 +08:00
services:
trellis:
2026-04-13 11:36:10 +08:00
image: zhouchengrong/fida-trellis-a6000:latest # 你 commit 的镜像
container_name: FiDA_3D_Trellis
2026-03-17 11:29:17 +08:00
restart: unless-stopped
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
volumes:
- .:/workspace # 当前目录挂载到 /workspace便于开发
ports:
2026-03-17 12:18:14 +08:00
- "14201:8120"
2026-03-17 11:29:17 +08:00
working_dir: /workspace
tty: true
stdin_open: true
# 最新GPU配置方式替代旧的环境变量声明更规范
runtime: nvidia
deploy:
resources:
reservations:
devices:
- driver: nvidia
2026-03-17 12:31:05 +08:00
device_ids: [ '1' ]
2026-03-17 12:29:50 +08:00
capabilities: [ gpu, compute, utility ] # 移除 video非必需减少兼容问题
2026-03-17 11:29:17 +08:00
command: >
bash -c "
2026-04-13 11:36:10 +08:00
conda run --no-capture-output -n trellis python server.py
"