Remove NIXL reinstall step (#51882)
Signed-off-by: Ovidiu Mara <ovidium@nvidia.com>
ovidiusm committed
Aug 13, 2026 at 07:44 UTC
5658391af7db332a46389738dad5dfe99f7c55a8
2 files changed
+1
-6
docker/Dockerfile
-5
@@ -1016,11 +1016,6 @@ RUN --mount=type=cache,target=/opt/uv/cache \
1016
# clean up -dev packages, keep runtime libraries
1017
rm -rf /var/lib/apt/lists/* \
1018
); \
1019
- # Force-reinstall the matching CUDA wheel so the correct nixl_ep_cpp.so is
1020
- # installed. Chained with && through the mooncake swap below: this RUN has no
1021
- # `set -e`, so its status is that of the last command, and a trailing no-op
1022
- # would otherwise mask a failure here.
1023
- uv pip install --system --force-reinstall --no-deps nixl-cu${CUDA_MAJOR} && \
1019
# The default mooncake wheel links libcudart.so.12, so on CUDA 13 swap in the
1020
# cuda13 variant. Uninstall first; both ship the `mooncake` package.
1021
# Mirrors .buildkite/scripts/install-kv-connectors.sh.
requirements/kv_connectors.txt
+1
-1
@@ -2,7 +2,7 @@ lmcache >= 0.3.9
2
# CuPy 14.1.0 imports pytest from cupy.testing._random. Use <14.1.0
3
# until a fixed newer release is verified for runtime images.
4
cupy-cuda13x < 14.1.0
5
-nixl == 1.3.1
5
+nixl == 1.3.2
6
# CUDA 12 build. On the CUDA 13 image install-kv-connectors.sh swaps this for
7
# the mooncake-transfer-engine-cuda13 variant of the same version.
8
mooncake-transfer-engine >= 0.3.12