chore: update Dockerfile (#9820)
This commit is contained in:
@@ -94,7 +94,8 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel html5li
|
|||||||
# Download source files
|
# 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 && \
|
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 && \
|
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 && \
|
tar -xf nvshmem_src_cuda12-all-all-3.3.9.tar.gz && \
|
||||||
mv nvshmem_src nvshmem && \
|
mv nvshmem_src nvshmem && \
|
||||||
rm -f /sgl-workspace/nvshmem_src_cuda12-all-all-3.3.9.tar.gz
|
rm -f /sgl-workspace/nvshmem_src_cuda12-all-all-3.3.9.tar.gz
|
||||||
|
|||||||
Reference in New Issue
Block a user