software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed missing Infiniband dependency, added CuDNN

Browse Source
This commit is contained in:
eugr
2025-12-14 21:49:50 -08:00
parent 33720fc9d6
commit 4551795908
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@@ -19,6 +19,7 @@ ENV VLLM_BASE_DIR=/workspace/vllm
RUN apt update && apt upgrade -y \ RUN apt update && apt upgrade -y \
&& apt install -y --allow-change-held-packages --no-install-recommends \ && apt install -y --allow-change-held-packages --no-install-recommends \
curl vim cmake build-essential ninja-build \ curl vim cmake build-essential ninja-build \
libcudnn9-cuda-13 libcudnn9-dev-cuda-13 \
python3-dev python3-pip git wget \ python3-dev python3-pip git wget \
libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \ libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
ccache \ ccache \
@@ -165,7 +166,8 @@ ENV PIP_CACHE_DIR=/root/.cache/pip
RUN apt update && apt upgrade -y \ RUN apt update && apt upgrade -y \
&& apt install -y --allow-change-held-packages --no-install-recommends \ && apt install -y --allow-change-held-packages --no-install-recommends \
python3 python3-pip python3-dev vim curl git wget \ python3 python3-pip python3-dev vim curl git wget \
libnccl-dev libnccl2 libibverbs1 rdma-core \ libcudnn9-cuda-13 \
libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# Set final working directory # Set final working directory