EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Minor fix docker container DeepEP on multi platforms (#9205)

Browse Source
This commit is contained in:
fzyzcjy
2025-08-15 08:41:49 +08:00
committed by GitHub
parent 004f7f1972
commit 392de007cb
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
docker/Dockerfile
View File

@@ -102,7 +102,18 @@ RUN cd /sgl-workspace/nvshmem && \
# Install DeepEP # Install DeepEP
RUN cd /sgl-workspace/DeepEP && \ RUN cd /sgl-workspace/DeepEP && \
TORCH_CUDA_ARCH_LIST='9.0;10.0' NVSHMEM_DIR=${NVSHMEM_DIR} pip install . case "$CUDA_VERSION" in \
12.6.1) \
CHOSEN_TORCH_CUDA_ARCH_LIST='9.0' \
;; \
12.8.1|12.9.1) \
CHOSEN_TORCH_CUDA_ARCH_LIST='9.0;10.0' \
;; \
*) \
echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1 \
;; \
esac && \
NVSHMEM_DIR=${NVSHMEM_DIR} TORCH_CUDA_ARCH_LIST="${CHOSEN_TORCH_CUDA_ARCH_LIST}" pip install .
# Python tools # Python tools
RUN python3 -m pip install --no-cache-dir \ RUN python3 -m pip install --no-cache-dir \