services:
  trellis:
    image: zhouchengrong/fida-trellis:latest  # 你 commit 的镜像
    container_name: trellis-dev
    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
              capabilities: [ gpu, compute, video ]
              device_ids: [ '1' ]
    command: >
      bash -c "
        /opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' &&
        tail -f /dev/null
      "