diff --git a/docker/Dockerfile b/docker/Dockerfile index 39ecdcd68e..0b68f89dde 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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. diff --git a/requirements/kv_connectors.txt b/requirements/kv_connectors.txt index 78b1964c3f..d7a396bfa0 100644 --- a/requirements/kv_connectors.txt +++ b/requirements/kv_connectors.txt @@ -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