x

2026-06-03 15:39:09 +08:00
parent 48a425af3e
commit b1b40b6242
5 changed files with 908 additions and 0 deletions
@@ -0,0 +1,71 @@
+# =============================================================================
+# MinerU ROCm Docker Compose 配置
+# 原生 Linux + AMD GPU 环境
+# =============================================================================
+
+services:
+  mineru:
+    image: mineru-rocm:7.2.1
+    build:
+      context: .
+      dockerfile: Dockerfile
+      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 权限
+    # 如果要与宿主机用户对齐，取消下面注释并改为你的 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"