Skip to content

Commit 6f082f6

Browse files
committed
nccl_2.11.4
1 parent 41bf393 commit 6f082f6

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

commons/Dockerfile.base.py38-cuda11.4

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,6 @@ RUN rm -fr /var/lib/apt/lists/* && \
5656
rm GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB && \
5757
wget https://apt.repos.intel.com/setup/intelproducts.list -O /etc/apt/sources.list.d/intelproducts.list && \
5858
apt-get update -y && \
59-
apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
60-
RUN apt-get update && \
6159
apt-get install -y --no-install-recommends \
6260
automake \
6361
bashtop \
@@ -162,7 +160,7 @@ RUN apt-get update && \
162160
unzip \
163161
v4l-utils \
164162
vim \
165-
x264
163+
x264 \
166164
xvfb \
167165
xz-utils \
168166
yasm \
@@ -175,9 +173,9 @@ RUN apt-get update && \
175173
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete
176174

177175
WORKDIR /tmp
178-
RUN curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.8/nccl_2.11.4-1+cuda11.4_x86_64.txz -O && \
179-
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.*' && \
180-
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/*' && \
176+
RUN curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.11/nccl_2.11.4-1+cuda11.4_x86_64.txz -O && \
177+
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.4_x86_64.txz -C /usr/local/cuda/lib64/ --strip-components=2 --wildcards '*/lib/libnccl.so.*' && \
178+
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.4_x86_64.txz -C /usr/lib/pkgconfig/ --strip-components=3 --wildcards '*/lib/pkgconfig/*' && \
181179
rm nccl_${NCCL}-1+cuda11.4_x86_64.txz && \
182180
ldconfig
183181

0 commit comments

Comments
 (0)