diff --git a/docker/Dockerfile b/docker/Dockerfile index 7b59adf..66e4a84 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -177,6 +177,7 @@ RUN set -ex && \ -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ -DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ -DHIP_PATH=/opt/rocm \ + -DCMAKE_INCLUDE_PATH=/opt/rocm/include \ -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \ cd /opt/vllm_build && ninja -j8 && \ cp /opt/vllm_build/*.abi3.so /opt/vllm/vllm/ && \