diff --git a/docker/Dockerfile b/docker/Dockerfile index cfca35f..ef1b48e 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -253,7 +253,8 @@ RUN set -ex && \ -DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ -DHIP_PATH=/opt/rocm \ -DHIP_INCLUDE_DIR=/opt/rocm/include \ - -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \ + -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" \ + -DCMAKE_LIBRARY_PATH=/opt/rocm/lib && \ cd /opt/vllm_build && ninja -j8 && \ cp /opt/vllm_build/*.abi3.so /opt/vllm/vllm/ && \ echo "vllm C++ build complete (layer cached)."