From 33156f56e0b1a4e872048669ae625fe3067f9c24 Mon Sep 17 00:00:00 2001
From: Shang Wang <shangw@nvidia.com>
Date: Thu, 8 Jan 2026 02:47:02 -0500
Subject: [PATCH] [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>
---
 docker/Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 0daff3995..ec6bfc5df 100644
--- 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%%.*}"; \