From 8229280a9c74a2f01d486dee4165eba881b5579d Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Mon, 6 Oct 2025 17:05:33 -0700 Subject: [PATCH] [Misc] Define EP kernel arch list in Dockerfile (#25635) Signed-off-by: Simon Mo --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 661846d3549ca..f9df931e73b17 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -461,7 +461,7 @@ RUN set -eux; \ # Install EP kernels(pplx-kernels and DeepEP) COPY tools/ep_kernels/install_python_libraries.sh install_python_libraries.sh ENV CUDA_HOME=/usr/local/cuda -RUN export TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-9.0a+PTX}" \ +RUN export TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-9.0a 10.0a+PTX}" \ && bash install_python_libraries.sh # CUDA image changed from /usr/local/nvidia to /usr/local/cuda in 12.8 but will