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
      "