diff --git a/Dockerfile b/Dockerfile index 9df6ab7..43bfe71 100644 --- a/Dockerfile +++ b/Dockerfile @@ -66,9 +66,9 @@ ENV EXT_PARALLEL=2 ENV NVCC_APPEND_FLAGS="--threads 2" ENV MAX_JOBS=4 # Target RunPod GPU architectures: -# 8.0 = A100, 8.6 = A10/RTX 3090, 8.9 = RTX 4090/L40, 9.0 = H100/H200 -# Note: Blackwell (10.0) not yet supported by SageAttention -ENV TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0" +# 8.0 = A100, 9.0 = H100/H200 +# Note: 8.6/8.9 excluded due to SageAttention sm90 kernel cross-compile issues +ENV TORCH_CUDA_ARCH_LIST="8.0;9.0" RUN git clone https://github.com/thu-ml/SageAttention.git && \ cd SageAttention && \ git checkout 2aecfa8 && \