diff --git a/Dockerfile b/Dockerfile index f407de0..3b74af4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -65,8 +65,9 @@ WORKDIR /tmp ENV EXT_PARALLEL=2 ENV NVCC_APPEND_FLAGS="--threads 2" ENV MAX_JOBS=4 -# Target RunPod GPU architectures: H100/H200(9.0), RTX 5090/Blackwell(10.0) -ENV TORCH_CUDA_ARCH_LIST="9.0;10.0" +# Target RunPod GPU architectures: H100/H200(9.0) +# Note: Blackwell (10.0) not yet supported by SageAttention +ENV TORCH_CUDA_ARCH_LIST="9.0" RUN git clone https://github.com/thu-ml/SageAttention.git && \ cd SageAttention && \ pip install --no-build-isolation . && \