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

services:
  mineru:
    image: mineru-rocm:7.2.1
    build:
      context: .
      dockerfile: Dockerfile
      network: host                       # 构建时走宿主机网络（国内访问 ROCm 源需要）
      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
        # ---- 代理（国内必需）----
        HTTP_PROXY: http://127.0.0.1:8118
        HTTPS_PROXY: http://127.0.0.1:8118
    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 权限
    # 如果要与宿主机用户对齐，取消下面注释并改为你的 UID/GID
    # user: "${UID:-1000}:${GID:-1000}"

    # ---- 环境变量 ----
    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，可改）----
    # 例如直接处理 PDF:
    #   command: mineru -p /data/input/example.pdf -o /data/output -b hybrid-auto-engine
    # 启动 WebUI:
    #   command: mineru-gradio --server-name 0.0.0.0 --server-port 7860
    # 启动 API:
    #   command: mineru-api --host 0.0.0.0 --port 8000
    command: bash

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

    restart: "no"