67 lines
2.3 KiB
YAML
67 lines
2.3 KiB
YAML
# =============================================================================
|
|
# 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
|
|
# ---- Git 代理（仅 GitHub 需要，国内无镜像）----
|
|
GIT_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 权限
|
|
|
|
# ---- 环境变量 ----
|
|
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"
|