28 lines
902 B
YAML
Executable File
28 lines
902 B
YAML
Executable File
services:
|
||
trellis:
|
||
image: zhouchengrong/fida-trellis-a6000:latest # 你 commit 的镜像
|
||
container_name: FiDA_3D_Trellis
|
||
restart: unless-stopped
|
||
environment:
|
||
- NVIDIA_VISIBLE_DEVICES=all
|
||
- NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
|
||
volumes:
|
||
- .:/workspace # 当前目录挂载到 /workspace，便于开发
|
||
ports:
|
||
- "14201:8120"
|
||
working_dir: /workspace
|
||
tty: true
|
||
stdin_open: true
|
||
# 最新GPU配置方式（替代旧的环境变量声明，更规范）
|
||
runtime: nvidia
|
||
deploy:
|
||
resources:
|
||
reservations:
|
||
devices:
|
||
- driver: nvidia
|
||
device_ids: [ '1' ]
|
||
capabilities: [ gpu, compute, utility ] # 移除 video（非必需，减少兼容问题）
|
||
command: >
|
||
bash -c "
|
||
conda run --no-capture-output -n trellis python server.py
|
||
" |