x

2026-06-05 13:35:14 +08:00
parent 270faac7b7
commit 9d0b4b4749
2 changed files with 31 additions and 100 deletions
@@ -9,56 +9,30 @@ services:
    build:
      context: .
      dockerfile: Dockerfile
-      network: host                       # 容器共享宿主机网络（构建时需要）
+      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_PROXY: http://127.0.0.1:8118
    container_name: mineru-rocm
    stdin_open: true
    tty: true
-    ipc: host                          # vllm 共享内存需要
-
-    # ---- GPU 设备透传（必需）----
+    ipc: host
    devices:
-      - /dev/kfd                       # ROCm KFD 内核驱动接口
-      - /dev/dri                       # GPU 渲染节点 (renderD*)
-
-    # ---- 安全配置（ROCm 需要）----
+      - /dev/kfd
+      - /dev/dri
    security_opt:
-      - seccomp=unconfined             # 允许 ROCm 系统调用
-
-    # ---- 用户权限（访问 GPU 设备）----
+      - seccomp=unconfined
    group_add:
-      - video                          # /dev/dri/render* 权限
-      - render                         # /dev/kfd 权限
-
-    # ---- 环境变量 ----
+      - video
+      - render
    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"