From 0f283a515a507adccc212db387d76dfda4045582 Mon Sep 17 00:00:00 2001 From: Kangyan-Zhou Date: Mon, 4 May 2026 12:46:38 -0700 Subject: [PATCH] [Docker] fix: install nixl stub alongside nixl-cuXX binary (#24369) Co-authored-by: Claude Opus 4.7 (1M context) --- docker/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 2dfd81228..e6884d3b5 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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