From b0b5059789c4a428d7c281277cf6ad853fef94dc Mon Sep 17 00:00:00 2001 From: Andrey Khalyavin Date: Fri, 28 Nov 2025 14:18:01 +0300 Subject: [PATCH] Add libraries needed for building VLLM wheel to the test docker image. Signed-off-by: Andrey Khalyavin --- docker/Dockerfile | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index eb7c105071c0..b5f743d24511 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -468,8 +468,15 @@ ENV UV_LINK_MODE=copy RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \ && echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections \ + && CUDA_VERSION_DASH=$(echo $CUDA_VERSION | cut -d. -f1,2 | tr '.' '-') \ && apt-get update -y \ - && apt-get install -y git + && apt-get install -y git \ + cuda-nvrtc-dev-${CUDA_VERSION_DASH} \ + libcublas-dev-${CUDA_VERSION_DASH} \ + libcurand-dev-${CUDA_VERSION_DASH} \ + libcusolver-dev-${CUDA_VERSION_DASH} \ + libcusparse-dev-${CUDA_VERSION_DASH} && \ + rm -rf /var/lib/apt/lists/* # install development dependencies (for testing) RUN --mount=type=cache,target=/root/.cache/uv \