kanshan/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-04-23 16:08:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update dockerfile

Browse Source
This commit is contained in:
xlliu-scitix 2025-12-21 10:48:36 +00:00
parent d5d08b99cc
commit eb04f2127a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.cuda12.x.ubuntu20.04
View File

@ -74,8 +74,8 @@ RUN apt-mark unhold libnccl2 libnccl-dev || true && \
RUN wget https://download.open-mpi.org/release/open-mpi/v${MPI_SERIES}/openmpi-${MPI_VERSION}.tar.gz && \
tar zxvf openmpi-${MPI_VERSION}.tar.gz && \
cd openmpi-${MPI_VERSION} && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda && \
make -j$(nproc) && make install && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda > /dev/null && \
make -j$(nproc) > /dev/null && make install && \
rm -rf /workspace/openmpi-${MPI_VERSION} /workspace/openmpi-${MPI_VERSION}.tar.gz
# -------------------------

4
docker/Dockerfile.cuda13.x.ubuntu22.04
View File

@ -71,8 +71,8 @@ RUN apt-get install -y --no-install-recommends \
RUN wget https://download.open-mpi.org/release/open-mpi/v${MPI_SERIES}/openmpi-${MPI_VERSION}.tar.gz && \
tar zxvf openmpi-${MPI_VERSION}.tar.gz && \
cd openmpi-${MPI_VERSION} && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda && \
make -j$(nproc) && make install && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda > /dev/null && \
make -j$(nproc) > /dev/null && make install && \
rm -rf /workspace/openmpi-${MPI_VERSION} /workspace/openmpi-${MPI_VERSION}.tar.gz
# -------------------------