[docker] A follow-up patch to fix #30913: [docker] install cuda13 version of lmcache and nixl (#31775)

Signed-off-by: Shang Wang <shangw@nvidia.com>
2026-01-08 02:47:02 -05:00
parent 107cf8e92f
commit 33156f56e0
1 changed files with 2 additions and 0 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -627,6 +627,8 @@ ARG PIP_EXTRA_INDEX_URL UV_EXTRA_INDEX_URL
 ENV UV_HTTP_TIMEOUT=500

 # install kv_connectors if requested
+ARG torch_cuda_arch_list='7.0 7.5 8.0 8.9 9.0 10.0 12.0'
+ENV TORCH_CUDA_ARCH_LIST=${torch_cuda_arch_list}
 RUN --mount=type=cache,target=/root/.cache/uv \
    --mount=type=bind,source=requirements/kv_connectors.txt,target=/tmp/kv_connectors.txt,ro \
    CUDA_MAJOR="${CUDA_VERSION%%.*}"; \