Skip to content

Commit 7f71816

Browse files
bbartelskhluu
authored andcommitted
[CI/Build] Fixes missing runtime dependencies (#29822)
Signed-off-by: bbartels <[email protected]> (cherry picked from commit 2d613de)
1 parent 339e84c commit 7f71816

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

docker/Dockerfile

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,12 @@ RUN CUDA_VERSION_DASH=$(echo $CUDA_VERSION | cut -d. -f1,2 | tr '.' '-') && \
364364
cuda-cudart-${CUDA_VERSION_DASH} \
365365
cuda-nvrtc-${CUDA_VERSION_DASH} \
366366
cuda-cuobjdump-${CUDA_VERSION_DASH} \
367-
libcublas-${CUDA_VERSION_DASH} && \
367+
# https://github.com/vllm-project/vllm/issues/29590
368+
libcurand-dev-${CUDA_VERSION_DASH} \
369+
libcublas-${CUDA_VERSION_DASH} \
370+
# Fixes nccl_allocator requiring nccl.h at runtime
371+
# https://github.com/vllm-project/vllm/blob/1336a1ea244fa8bfd7e72751cabbdb5b68a0c11a/vllm/distributed/device_communicators/pynccl_allocator.py#L22
372+
libnccl-dev && \
368373
rm -rf /var/lib/apt/lists/*
369374

370375
ARG PIP_INDEX_URL UV_INDEX_URL

0 commit comments

Comments
 (0)