76 lines
2.7 KiB
YAML
76 lines
2.7 KiB
YAML
# =============================================================================
|
|
# 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"
|