# =============================================================================
# MinerU ROCm Docker Compose 配置
# 原生 Linux + AMD GPU 环境
# =============================================================================

services:
  mineru:
    image: mineru-rocm:7.2.1
    build:
      context: .
      dockerfile: Dockerfile
      network: host                       # 容器共享宿主机网络（构建时需要）
      args:
        # ---- 按你的 GPU 修改 ----
        # gfx1201 = RX 9070 / 9070 XT / 9070 GRE
        # gfx1200 = RX 9060 XT / 9060 XT LP
        # gfx1100 = RX 7900 XTX / XT / GRE
        # gfx1101 = RX 7800 XT / 7700 XT
        # gfx1030 = RX 6950 / 6900 / 6800 系列
        ARCH: gfx1201
    container_name: mineru-rocm
    stdin_open: true
    tty: true
    ipc: host                          # vllm 共享内存需要

    # ---- GPU 设备透传（必需）----
    devices:
      - /dev/kfd                       # ROCm KFD 内核驱动接口
      - /dev/dri                       # GPU 渲染节点 (renderD*)

    # ---- 安全配置（ROCm 需要）----
    security_opt:
      - seccomp=unconfined             # 允许 ROCm 系统调用

    # ---- 用户权限（访问 GPU 设备）----
    group_add:
      - video                          # /dev/dri/render* 权限
      - render                         # /dev/kfd 权限

    # ---- 环境变量 ----
    environment:
      - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface}
      - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface}
      - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope}

    # ---- 卷挂载 ----
    volumes:
      # 输入/输出目录（按需修改）
      - ${INPUT_DIR:-./data/input}:/data/input:ro
      - ${OUTPUT_DIR:-./data/output}:/data/output
      # 模型缓存（持久化，避免每次下载）
      - ${MODEL_DIR:-./data/models}:/opt/models
      # MIOpen kernel 缓存（持久化，避免每次预热）
      - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen

    # ---- 启动命令（默认 bash，可改）----
    command: bash

    # ---- 端口（WebUI / API 模式时启用）----
    # ports:
    #   - "7860:7860"   # WebUI
    #   - "8000:8000"   # API

    restart: "no"