39 lines
1.1 KiB
YAML
39 lines
1.1 KiB
YAML
# =============================================================================
|
|
# MinerU ROCm Docker Compose 配置
|
|
# 原生 Linux + AMD GPU 环境
|
|
# =============================================================================
|
|
|
|
services:
|
|
mineru:
|
|
image: mineru-rocm:7.2.1
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
network: host
|
|
args:
|
|
ARCH: gfx1201
|
|
GIT_PROXY: http://127.0.0.1:8118
|
|
container_name: mineru-rocm
|
|
stdin_open: true
|
|
tty: true
|
|
ipc: host
|
|
devices:
|
|
- /dev/kfd
|
|
- /dev/dri
|
|
security_opt:
|
|
- seccomp=unconfined
|
|
group_add:
|
|
- 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_CACHE:-./data/miopen}:/root/.cache/miopen
|
|
command: bash
|
|
restart: "no"
|