Skip to content

Commit f4041cb

Browse files
committed
nccl 2.8.3
1 parent b1f62ac commit f4041cb

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

commons/Dockerfile.base.cuda11.1

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
9090
libboost-filesystem-dev \
9191
libopenblas-dev liblapack-dev \
9292
xvfb \
93+
xz-utils \
9394
fonts-nanum \
9495
fonts-nanum-coding \
9596
fonts-nanum-extra \
@@ -100,9 +101,9 @@ RUN rm -fr /var/lib/apt/lists/* && \
100101
echo "/opt/intel/mkl/lib/intel64" >> /etc/ls.so.conf && \
101102
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete
102103

103-
RUN apt update && apt install curl xz-utils -y --no-install-recommends && NCCL_DOWNLOAD_SUM=34000cbe6a0118bfd4ad898ebc5f59bf5d532bbf2453793891fa3f1621e25653 && \
104-
curl -fsSL https://developer.nvidia.com/compute/machine-learning/nccl/secure/${NCCL}/ubuntu1804/x86_64/nccl-repo-ubuntu1804-${NCCL}-ga-cuda11.1_1-1_amd64.deb https://developer.nvidia.com/compute/machine-learning/nccl/secure/${NCCL}/agnostic/x64/nccl_${NCCL}-1+cuda11.1_x86_64.txz -O && \
105-
echo "$NCCL_DOWNLOAD_SUM nccl_2.8.3-1+cuda11.1_x86_64.txz" | sha256sum -c - && \
104+
# NCCL_DOWNLOAD_SUM=34000cbe6a0118bfd4ad898ebc5f59bf5d532bbf2453793891fa3f1621e25653
105+
# echo "$NCCL_DOWNLOAD_SUM nccl_2.8.3-1+cuda11.1_x86_64.txz" | sha256sum -c -
106+
RUN curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.8/nccl_2.8.3-1+cuda11.1_x86_64.txz -O && \
106107
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.1_x86_64.txz -C /usr/local/cuda/lib64/ --strip-components=2 --wildcards '*/lib/libnccl.so.*' && \
107108
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.1_x86_64.txz -C /usr/lib/pkgconfig/ --strip-components=3 --wildcards '*/lib/pkgconfig/*' && \
108109
rm nccl_${NCCL}-1+cuda11.1_x86_64.txz && \

0 commit comments

Comments
 (0)