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 . && \