experiments
/
STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bump NCCL version due to NVIDIA base image update

This commit is contained in:
Reuben Morais 2021-01-18 13:37:13 +00:00
parent 79a42b345d
commit 64465cd93a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.build.tmpl
View File

@ -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_CUDA_VERSION 10.1
ENV TF_CUDNN_VERSION 7.6 ENV TF_CUDNN_VERSION 7.6
ENV TF_CUDA_COMPUTE_CAPABILITIES 6.0 ENV TF_CUDA_COMPUTE_CAPABILITIES 6.0
ENV TF_NCCL_VERSION 2.7 ENV TF_NCCL_VERSION 2.8
# Common Environment Setup # Common Environment Setup
ENV TF_BUILD_CONTAINER_TYPE GPU ENV TF_BUILD_CONTAINER_TYPE GPU