From 9c99949ef3d92d2c0587c1fca6fd772b3634d357 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Sat, 30 Aug 2025 03:08:14 -0700 Subject: [PATCH] chore: update Dockerfile (#9820) --- docker/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 81a6b352e..e08c77b0e 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -94,7 +94,8 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel html5li # Download source files RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.3.9/source/nvshmem_src_cuda12-all-all-3.3.9.tar.gz && \ git clone https://github.com/deepseek-ai/DeepEP.git && \ - cd DeepEP && git checkout ${DEEPEP_COMMIT} && cd .. && \ + cd DeepEP && git checkout ${DEEPEP_COMMIT} && sed -i 's/#define NUM_CPU_TIMEOUT_SECS 100/#define NUM_CPU_TIMEOUT_SECS 1000/' csrc/kernels/configs.cuh && \ + cd .. && \ tar -xf nvshmem_src_cuda12-all-all-3.3.9.tar.gz && \ mv nvshmem_src nvshmem && \ rm -f /sgl-workspace/nvshmem_src_cuda12-all-all-3.3.9.tar.gz