Revert "Stop pinning nccl version. (#8421)"

ezyang · ezyang · commit c0972e14902a · 2018-06-13T11:53:21.000-04:00
This reverts commit 3cb45ba.
diff --git a/docker/caffe2/jenkins/common/install_nccl.sh b/docker/caffe2/jenkins/common/install_nccl.sh
@@ -30,6 +30,9 @@ if [ -n "$NCCL_UBUNTU_VER" ]; then
   apt-get install -y wget
   dpkg -i "${NCCL_DEB}"
 
+  # On March 8, 2018 Nvidia began recommending version 2.1.15
+  NCCL_LIB_VERSION="2.1.15-1+cuda${CUDA_VERSION:0:3}"
+
   apt update
-  apt install -y libnccl2 libnccl-dev
+  apt install -y libnccl2=$NCCL_LIB_VERSION libnccl-dev=$NCCL_LIB_VERSION
 fi
