Commit 1fca3282 authored by Felix Abecassis US's avatar Felix Abecassis US

Update to latest NCCL (2.3.5)

parent ae5083b3
Pipeline #33906695 passed with stages
in 14 minutes and 49 seconds
......@@ -7,7 +7,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-nvml-dev-$CUDA_PKG_VERSION \
cuda-minimal-build-$CUDA_PKG_VERSION \
cuda-command-line-tools-$CUDA_PKG_VERSION \
libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
libnccl-dev=$NCCL_VERSION-2+cuda10.0 && \
rm -rf /var/lib/apt/lists/*
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
......@@ -2,11 +2,11 @@ ARG repository
FROM ${repository}:10.0-base-ubuntu18.04
LABEL maintainer "NVIDIA CORPORATION <cudatools@nvidia.com>"
ENV NCCL_VERSION 2.3.4
ENV NCCL_VERSION 2.3.5
RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-libraries-$CUDA_PKG_VERSION \
cuda-nvtx-$CUDA_PKG_VERSION \
libnccl2=$NCCL_VERSION-1+cuda10.0 && \
libnccl2=$NCCL_VERSION-2+cuda10.0 && \
apt-mark hold libnccl2 && \
rm -rf /var/lib/apt/lists/*
......@@ -7,7 +7,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-nvml-dev-$CUDA_PKG_VERSION \
cuda-minimal-build-$CUDA_PKG_VERSION \
cuda-command-line-tools-$CUDA_PKG_VERSION \
libnccl-dev=$NCCL_VERSION-1+cuda9.2 && \
libnccl-dev=$NCCL_VERSION-2+cuda9.2 && \
rm -rf /var/lib/apt/lists/*
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
......@@ -2,11 +2,11 @@ ARG repository
FROM ${repository}:9.2-base-ubuntu18.04
LABEL maintainer "NVIDIA CORPORATION <cudatools@nvidia.com>"
ENV NCCL_VERSION 2.3.4
ENV NCCL_VERSION 2.3.5
RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-libraries-$CUDA_PKG_VERSION \
cuda-nvtx-$CUDA_PKG_VERSION \
libnccl2=$NCCL_VERSION-1+cuda9.2 && \
libnccl2=$NCCL_VERSION-2+cuda9.2 && \
apt-mark hold libnccl2 && \
rm -rf /var/lib/apt/lists/*
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment