File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -54,9 +54,9 @@ COPY . /vec-inf
5454RUN 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
6060ENV NCCL_DEBUG=INFO
6161
6262# Set the default command to start an interactive shell
Original file line number Diff line number Diff 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]
You can’t perform that action at this time.
0 commit comments