diff --git a/docker/Dockerfile b/docker/Dockerfile index 1962368..dc46ab6 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -238,6 +238,7 @@ RUN set -ex && \ mkdir -p /opt/vllm_build && \ export HIP_PLATFORM=amd HIP_PATH=/opt/rocm ROCM_PATH=/opt/rocm && \ export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ + export LIBRARY_PATH=/opt/rocm/lib:$LIBRARY_PATH && \ cmake -S /opt/vllm -B /opt/vllm_build -G Ninja \ -DCMAKE_BUILD_TYPE=RelWithDebInfo \ -DVLLM_TARGET_DEVICE=rocm \