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

fix allow-change-held-packages for cuda12.9 Dockerfile

Browse Source
This commit is contained in:
xlliu 2026-01-28 07:46:52 +00:00
parent 613b80389c
commit 63cb76d463
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.cuda12.9.ubuntu22.04
View File

@ -60,8 +60,10 @@ RUN { wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x
# 3. Install NCCL (pinned version)
# -------------------------
RUN { apt-get install -y --no-install-recommends \
--allow-change-held-packages \
libnccl2=${NCCL_PACKAGE_VERSION} \
libnccl-dev=${NCCL_PACKAGE_VERSION} > build.log 2>&1 && \
apt-mark hold libnccl2 libnccl-dev >> build.log 2>&1 && \
ldconfig >> build.log 2>&1 && \
rm -rf /var/lib/apt/lists/* && rm -f build.log; } || (cat build.log && false)
@ -126,7 +128,7 @@ RUN cd /usr/local/sihpc/lib && \
###########################
# Package Stage
###########################
FROM ubuntu:20.04 AS package
FROM ubuntu:${UBUNTU_VERSION} AS package
# Re-declare args for this stage (values are inherited)
ARG UBUNTU_VERSION

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

@ -126,7 +126,7 @@ RUN cd /usr/local/sihpc/lib && \
###########################
# Package Stage
###########################
FROM ubuntu:20.04 AS package
FROM ubuntu:${UBUNTU_VERSION} AS package
# Re-declare args for this stage (values are inherited)
ARG UBUNTU_VERSION