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 12:05:00 +00:00
parent a63433ce70
commit 0b7aeaf09b
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,7 +47,7 @@ RUN mv /etc/apt/sources.list.d/cuda*.list /tmp/disabled-cuda.list || true && \
apt-get install -y --no-install-recommends \
build-essential gcc g++ curl git wget ca-certificates \
make automake autoconf libtool pkg-config \
python3 python3-pip gzip xz-utils makeself && \
python3 python3-pip gzip xz-utils makeself > /dev/null 2>&1 && \
rm -rf /var/lib/apt/lists/*
# -------------------------
@ -55,7 +55,7 @@ RUN mv /etc/apt/sources.list.d/cuda*.list /tmp/disabled-cuda.list || true && \
# -------------------------
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb && \
dpkg -i cuda-keyring_1.1-1_all.deb && \
apt-get update
apt-get update > /dev/null 2>&1
# -------------------------
# 3. Install NCCL (pinned version)
@ -63,7 +63,7 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86
RUN apt-mark unhold libnccl2 libnccl-dev || true && \
apt-get install -y --no-install-recommends \
libnccl2=${NCCL_PACKAGE_VERSION} \
libnccl-dev=${NCCL_PACKAGE_VERSION} && \
libnccl-dev=${NCCL_PACKAGE_VERSION} > /dev/null 2>&1 && \
apt-mark hold libnccl2 libnccl-dev && \
ldconfig && \
rm -rf /var/lib/apt/lists/*
@ -72,7 +72,7 @@ RUN apt-mark unhold libnccl2 libnccl-dev || true && \
# 4. Build OpenMPI from source
# -------------------------
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 && \
tar zxvf openmpi-${MPI_VERSION}.tar.gz > /dev/null 2>&1 && \
cd openmpi-${MPI_VERSION} && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda > /dev/null 2>&1 && \
make -j$(nproc) > /dev/null 2>&1 && make install && \

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

@ -46,7 +46,7 @@ RUN apt-get -o Acquire::http::No-Cache=true update && \
apt-get install -y --no-install-recommends \
build-essential gcc g++ curl git wget ca-certificates \
make automake autoconf libtool pkg-config \
python3 python3-pip gzip xz-utils makeself && \
python3 python3-pip gzip xz-utils makeself > /dev/null 2>&1 && \
rm -rf /var/lib/apt/lists/*
# -------------------------
@ -54,14 +54,14 @@ RUN apt-get -o Acquire::http::No-Cache=true update && \
# -------------------------
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb && \
dpkg -i cuda-keyring_1.1-1_all.deb && \
apt-get update
apt-get update > /dev/null 2>&1
# -------------------------
# 3. Install NCCL (pinned version)
# -------------------------
RUN apt-get install -y --no-install-recommends \
libnccl2=${NCCL_PACKAGE_VERSION} \
libnccl-dev=${NCCL_PACKAGE_VERSION} && \
libnccl-dev=${NCCL_PACKAGE_VERSION} > /dev/null 2>&1 && \
ldconfig && \
rm -rf /var/lib/apt/lists/*
@ -69,7 +69,7 @@ RUN apt-get install -y --no-install-recommends \
# 4. Build OpenMPI from source
# -------------------------
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 && \
tar zxvf openmpi-${MPI_VERSION}.tar.gz > /dev/null 2>&1 && \
cd openmpi-${MPI_VERSION} && \
./configure --prefix=/usr/local/sihpc --with-cuda=/usr/local/cuda > /dev/null 2>&1 && \
make -j$(nproc) > /dev/null 2>&1 && make install && \