[Docker] fix: install nixl stub alongside nixl-cuXX binary (#24369)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Kangyan-Zhou
2026-05-04 12:46:38 -07:00
committed by GitHub
co-authored by Claude Opus 4.7
parent 60a1dacd89
commit 0f283a515a
+2 -2
View File
@@ -540,10 +540,10 @@ RUN --mount=type=cache,target=/root/.cache/pip \
# it with --no-deps and pair it with the matching nixl-cu12 / nixl-cu13 binary
# to avoid shipping wrong-CUDA libs on cu13 images.
RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
python3 -m pip install nixl-cu12 --no-deps ; \
python3 -m pip install nixl nixl-cu12 --no-deps ; \
python3 -m pip install cuda-python==12.9 ; \
elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \
python3 -m pip install nixl-cu13 --no-deps ; \
python3 -m pip install nixl nixl-cu13 --no-deps ; \
python3 -m pip install cuda-python==13.2.0 ; \
fi