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 2025-12-21 10:18:09 +08:00
parent 4aebbcbf93
commit 7dce185a16
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Dockerfile
View File

@ -137,10 +137,9 @@ ENV NCCL_PACKAGE_VERSION=${NCCL_PACKAGE_VERSION} \
MPI_VERSION=${MPI_VERSION} \
BUILD_DATE=${BUILD_DATE}
WORKDIR /build
COPY --from=build /usr/local/sihpc /usr/local/sihpc
WORKDIR /build
WORKDIR /dist
RUN apt-get update && apt-get install -y --no-install-recommends makeself && \
chmod +x /usr/local/sihpc/bin/install_sihpc && \
SAFE_NCCL_PKG="${NCCL_PACKAGE_VERSION//+/-}" && \
@ -148,6 +147,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends makeself && \
makeself --gzip /usr/local/sihpc \
"${PACKAGE_FILENAME}" \
"SiHPC MPI + NCCL + NCCL-tests Portable Installer" \
./bin/install_sihpc
./bin/install_sihpc && \
rm -rf /usr/local/sihpc && \
apt-get autoremove -y && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
CMD ["bash", "-c", "SAFE_NCCL_PKG=${NCCL_PACKAGE_VERSION//+/-}; FILE=\"sicl-nccl${SAFE_NCCL_PKG}-ompi${MPI_VERSION}-ubuntu${UBUNTU_VERSION}-${BUILD_DATE}.run\"; ls -lh \"/build/$FILE\" && echo 'Build complete.'"]