File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -90,6 +90,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
90
90
libboost-filesystem-dev \
91
91
libopenblas-dev liblapack-dev \
92
92
xvfb \
93
+ xz-utils \
93
94
fonts-nanum \
94
95
fonts-nanum-coding \
95
96
fonts-nanum-extra \
@@ -100,9 +101,9 @@ RUN rm -fr /var/lib/apt/lists/* && \
100
101
echo "/opt/intel/mkl/lib/intel64" >> /etc/ls.so.conf && \
101
102
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete
102
103
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 && \
106
107
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.*' && \
107
108
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/*' && \
108
109
rm nccl_${NCCL}-1+cuda11.1_x86_64.txz && \
You can’t perform that action at this time.
0 commit comments