29 lines
961 B
YAML
29 lines
961 B
YAML
|
services:
|
|||
|
|
trellis:
|
|||
|
|
image: my-trellis-with-blender:20260317 # 你 commit 的镜像
|
|||
|
|
container_name: trellis-dev
|
|||
|
|
restart: unless-stopped
|
|||
|
|
environment:
|
|||
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|||
|
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
|
|||
|
|
volumes:
|
|||
|
|
- .:/workspace # 当前目录挂载到 /workspace，便于开发
|
|||
|
|
ports:
|
|||
|
|
- "7412:8120"
|
|||
|
|
working_dir: /workspace
|
|||
|
|
tty: true
|
|||
|
|
stdin_open: true
|
|||
|
|
# 最新GPU配置方式（替代旧的环境变量声明，更规范）
|
|||
|
|
runtime: nvidia
|
|||
|
|
deploy:
|
|||
|
|
resources:
|
|||
|
|
reservations:
|
|||
|
|
devices:
|
|||
|
|
- driver: nvidia
|
|||
|
|
count: all # 使用所有可用GPU，也可指定数量如count: 1
|
|||
|
|
capabilities: [ gpu, compute, video ]
|
|||
|
|
command: >
|
|||
|
|
bash -c "
|
|||
|
|
/opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' &&
|
|||
|
|
tail -f /dev/null
|
|||
|
|
"
|