Skip to content

Commit cc850d8

Browse files
committed
Revert to use vllm nccl 2.12.1 for now
1 parent c18aabb commit cc850d8

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ COPY . /vec-inf
5454
RUN PIP_INDEX_URL="https://download.pytorch.org/whl/cu128" uv pip install --system -e .[dev]
5555

5656
# Final configuration
57-
# RUN mkdir -p /vec-inf/nccl && \
58-
# mv /root/.config/vllm/nccl/cu12/libnccl.so.2.18.1 /vec-inf/nccl/libnccl.so.2.18.1
59-
# ENV VLLM_NCCL_SO_PATH=/vec-inf/nccl/libnccl.so.2.18.1
57+
RUN mkdir -p /vec-inf/nccl && \
58+
mv /root/.config/vllm/nccl/cu12/libnccl.so.2.18.1 /vec-inf/nccl/libnccl.so.2.18.1
59+
ENV VLLM_NCCL_SO_PATH=/vec-inf/nccl/libnccl.so.2.18.1
6060
ENV NCCL_DEBUG=INFO
6161

6262
# Set the default command to start an interactive shell

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ dev = [
4242
"xgrammar>=0.1.11",
4343
"torch>=2.7.0",
4444
"vllm>=0.10.0",
45+
"vllm-nccl-cu12>=2.18,<2.19",
4546
"ray==2.43.0",
4647
"cupy-cuda12x==12.1.0"
4748
]

0 commit comments

Comments
 (0)