x

2026-06-17 14:58:44 +08:00
parent f3ea5b2dbf
commit a05af55271
1 changed files with 9 additions and 1 deletions
@@ -25,7 +25,15 @@ ENV DEBIAN_FRONTEND=noninteractive \
    MINERU_MODEL_SOURCE=huggingface \
    TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=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