From 7a8b59f471f586089f4dfbd590ebf46a058e0492 Mon Sep 17 00:00:00 2001 From: Nick Date: Fri, 26 Dec 2025 12:11:31 +1300 Subject: [PATCH] Target H200 and RTX 5090 GPU architectures MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 448a955..4a01b13 100644 --- a/Dockerfile +++ b/Dockerfile @@ -65,8 +65,8 @@ WORKDIR /tmp ENV EXT_PARALLEL=4 ENV NVCC_APPEND_FLAGS="--threads 8" ENV MAX_JOBS=32 -# Target RunPod GPU architectures: A100(8.0), RTX 30xx(8.6), RTX 40xx(8.9), H100(9.0) -ENV TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0" +# Target RunPod GPU architectures: H100/H200(9.0), RTX 5090/Blackwell(10.0) +ENV TORCH_CUDA_ARCH_LIST="9.0;10.0" RUN git clone https://github.com/thu-ml/SageAttention.git && \ cd SageAttention && \ pip install --no-build-isolation . && \