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

Remove TF_NCCL_VERSION from Dockerfile.gpu.ppc64le

Browse Source 
Now NCCL2 is available in open source and TensorFlow is building
with it as the default there is no need to force the version to 1
This commit is contained in:
William D. Irons 2018-10-18 17:56:01 -04:00
parent 441c7775c1
commit 0cdab33274
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le
View File

@ -28,6 +28,3 @@ ENV TF_NEED_CUDA 1
ENV TF_CUDA_COMPUTE_CAPABILITIES 3.0
ENV TF_CUDA_VERSION 9.2
ENV CUDA_TOOLKIT_PATH /usr/local/cuda-9.2
# TODO get NCCL 2 in the docker image
ENV TF_NCCL_VERSION 1