diff --git a/docker/Dockerfile b/docker/Dockerfile index dc46ab6..4716e13 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -251,7 +251,8 @@ RUN set -ex && \ -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_LIBRARY_PATH=/opt/rocm/lib && \ + -DCMAKE_SHARED_LINKER_FLAGS="-L/opt/rocm/lib" \ + -DCMAKE_EXE_LINKER_FLAGS="-L/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)."