Files
mineru-rocm/docker/env.example
T
2026-06-11 18:10:02 +08:00

30 lines
970 B
Plaintext

# Docker 环境变量(按需修改)
# 复制为 .env 后使用: cp env.example .env
# ---- 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
# ---- 目录挂载 ----
INPUT_DIR=./data/input
OUTPUT_DIR=./data/output
MODEL_DIR=./data/models
MIOPEN_CACHE=./data/miopen
# ---- 默认 WebUI 使用的 GPU ----
# 单卡保持 0;多卡时可指定 0/1/...,或留空让 ROCm 使用全部可见 GPU。
HIP_VISIBLE_DEVICES=0
# ---- 可选:router-api profile 使用的 Worker 地址 ----
# 单 worker: http://mineru-worker0:8001
# 双 worker: http://mineru-worker0:8001,http://mineru-worker1:8002
ROUTER_API_URLS=http://mineru-worker0:8001
# ---- 模型下载源 ----
# huggingface(默认,需科学上网)或 modelscope(国内可用)
MINERU_MODEL_SOURCE=huggingface