diff --git a/docker/Dockerfile b/docker/Dockerfile index 89f8696..13c300d 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -188,6 +188,7 @@ RUN set -ex && \ -DHIP_ROOT_DIR=/opt/rocm \ -DROCM_PATH=/opt/rocm \ -DCMAKE_HIP_ARCHITECTURES=${ARCH} \ + -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \ # ninja 编译(-j8,32GB 内存;若 < 16GB 请改为 -j2) cd /opt/vllm_build && ninja -j8 && \