From a05af55271dd48e2182736943648d96c8780e124 Mon Sep 17 00:00:00 2001
From: chenjw28 <792430652@qq.com>
Date: Wed, 17 Jun 2026 14:58:44 +0800
Subject: [PATCH] x

---
 docker/Dockerfile | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 9a48ab7..95074ef 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -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