mineru-rocm/docker/docker-compose.yml

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

services:
  # --- WebUI 前端（纯前端，不本地处理，通过 Router → Worker 处理）---
  gradio:
    image: mineru-rocm:7.2.1
    profiles: ["gradio"]
    container_name: mineru-gradio
    stdin_open: true
    tty: true
    ipc: host
    ports:
      - "10002:7860"
    environment:
      - GRADIO_SERVER_NAME=0.0.0.0
      - MINERU_API_BASE=http://mineru-router:8000
      - 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
      - ./scripts:/opt/scripts:ro
    command:
      [
        "/opt/mineru_venv/bin/python", "/opt/scripts/gradio_client.py",
      ]
    depends_on:
      - router

  # --- 双卡 Worker（GPU 算力，无 profile，始终可用）---
  worker0:
    image: mineru-rocm:7.2.1
    build:
      context: .
      dockerfile: Dockerfile
      network: host
      args:
        ARCH: ${ARCH:-gfx1201}
        GIT_PROXY: ${GIT_PROXY:-}
    container_name: mineru-worker0
    stdin_open: true
    tty: true
    ipc: host
    devices:
      - /dev/kfd
      - /dev/dri
    security_opt:
      - seccomp=unconfined
    group_add:
      - video
    environment:
      - HIP_VISIBLE_DEVICES=0
      - PYTHONPATH=/opt/vllm:${PYTHONPATH:-}
      - 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_CACHE:-./data/miopen}:/root/.cache/miopen
      - ./scripts:/opt/scripts:ro
    command: ["mineru-api", "--host", "0.0.0.0", "--port", "8001", "--allow-public-http-client"]

  worker1:
    image: mineru-rocm:7.2.1
    container_name: mineru-worker1
    stdin_open: true
    tty: true
    ipc: host
    devices:
      - /dev/kfd
      - /dev/dri
    security_opt:
      - seccomp=unconfined
    group_add:
      - video
    environment:
      - HIP_VISIBLE_DEVICES=1
      - PYTHONPATH=/opt/vllm:${PYTHONPATH:-}
      - 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_CACHE:-./data/miopen}:/root/.cache/miopen
      - ./scripts:/opt/scripts:ro
    command: ["mineru-api", "--host", "0.0.0.0", "--port", "8002", "--allow-public-http-client"]

  router:
    image: mineru-rocm:7.2.1
    container_name: mineru-router
    stdin_open: true
    tty: true
    ipc: host
    ports:
      - "8000:8000"
    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
      - ./scripts:/opt/scripts:ro
    command:
      [
        "mineru-router",
        "--api-urls",
        "http://mineru-worker0:8001,http://mineru-worker1:8002",
        "--host", "0.0.0.0",
        "--port", "8000",
        "--allow-public-http-client",
      ]
    depends_on:
      - worker0
      - worker1