Skip to content

GitLab

Unverified Commit b3363cc1 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Fix docker container DeepEP error on Blackwell (#9171)

parent 98457c04
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile
View file @ b3363cc1
...@@ -102,7 +102,7 @@ RUN cd /sgl-workspace/nvshmem && \ ...@@ -102,7 +102,7 @@ RUN cd /sgl-workspace/nvshmem && \
# Install DeepEP # Install DeepEP
RUN cd /sgl-workspace/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 # Python tools
RUN python3 -m pip install --no-cache-dir \ RUN python3 -m pip install --no-cache-dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment