Skip to content

Commit b0b5059

Browse files
committed
Add libraries needed for building VLLM wheel to the test docker image.
Signed-off-by: Andrey Khalyavin <halyavin@yandex-team.ru>
1 parent 5c2b5cb commit b0b5059

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

docker/Dockerfile

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,8 +468,15 @@ ENV UV_LINK_MODE=copy
468468

469469
RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
470470
&& echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections \
471+
&& CUDA_VERSION_DASH=$(echo $CUDA_VERSION | cut -d. -f1,2 | tr '.' '-') \
471472
&& apt-get update -y \
472-
&& apt-get install -y git
473+
&& apt-get install -y git \
474+
cuda-nvrtc-dev-${CUDA_VERSION_DASH} \
475+
libcublas-dev-${CUDA_VERSION_DASH} \
476+
libcurand-dev-${CUDA_VERSION_DASH} \
477+
libcusolver-dev-${CUDA_VERSION_DASH} \
478+
libcusparse-dev-${CUDA_VERSION_DASH} && \
479+
rm -rf /var/lib/apt/lists/*
473480

474481
# install development dependencies (for testing)
475482
RUN --mount=type=cache,target=/root/.cache/uv \

0 commit comments

Comments
 (0)