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