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