Fix docker container DeepEP error on Blackwell (#9171)

b3363cc1 · fzyzcjy · GitHub · 98457c04 · b3363cc1
Unverified Commit b3363cc1 authored Aug 14, 2025 by fzyzcjy Committed by GitHub Aug 13, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docker/Dockerfile docker/Dockerfile +1 -1

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -102,7 +102,7 @@ RUN cd /sgl-workspace/nvshmem && \

 # Install DeepEP
 RUN cd /sgl-workspace/DeepEP && \
-    NVSHMEM_DIR=${NVSHMEM_DIR} pip install .
+    TORCH_CUDA_ARCH_LIST='9.0;10.0' NVSHMEM_DIR=${NVSHMEM_DIR} pip install .

 # Python tools
 RUN python3 -m pip install --no-cache-dir \