From 62c1075681d8ffd60b6fb1febd43c0b3b3d4f013 Mon Sep 17 00:00:00 2001 From: Alexandre Lissy Date: Tue, 15 Sep 2020 13:40:12 +0200 Subject: [PATCH] Fix #3321: Update NCCL dep to 2.7 following NVIDIA update --- Dockerfile.build.tmpl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.build.tmpl b/Dockerfile.build.tmpl index 1e4e2c57..b6bf3938 100644 --- a/Dockerfile.build.tmpl +++ b/Dockerfile.build.tmpl @@ -64,7 +64,7 @@ ENV TF_CUDA_PATHS "/usr,/usr/local/cuda-10.1,/usr/lib/x86_64-linux-gnu/" ENV TF_CUDA_VERSION 10.1 ENV TF_CUDNN_VERSION 7.6 ENV TF_CUDA_COMPUTE_CAPABILITIES 6.0 -ENV TF_NCCL_VERSION 2.4 +ENV TF_NCCL_VERSION 2.7 # Common Environment Setup ENV TF_BUILD_CONTAINER_TYPE GPU