cjw/mineru-rocm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
chenjw28
2026-06-17 14:58:44 +08:00
parent f3ea5b2dbf
commit a05af55271
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+9 -1
View File
@@ -25,7 +25,15 @@ ENV DEBIAN_FRONTEND=noninteractive \
MINERU_MODEL_SOURCE=huggingface \ MINERU_MODEL_SOURCE=huggingface \
TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 \ TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 \
HSA_ENABLE_SDMA=1 \ HSA_ENABLE_SDMA=1 \
VLLM_TARGET_DEVICE=rocm VLLM_TARGET_DEVICE=rocm \
CMAKE_PREFIX_PATH=/opt/rocm \
CMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
HIP_PATH=/opt/rocm \
HIP_ROOT_DIR=/opt/rocm \
HIP_INCLUDE_DIR=/opt/rocm/include \
HIP_PLATFORM=amd \
ROCM_PATH=/opt/rocm
WORKDIR /opt WORKDIR /opt