message: Link NCCL 2.x lib and header where ./configure expects them.
diff: @@ -35,6 +35,15 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
  find /usr/local/cuda-9.0/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
  rm /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a
  +# Link NCCL libray and header where the build script expects them.
  +RUN mkdir /usr/local/cuda-9.0/lib && \
  + ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/lib/libnccl.so.2 && \
  + ln -s /usr/include/nccl.h /usr/local/cuda/include/nccl.h
  +
  +# TODO(tobyboyd): Remove after license is excluded from BUILD file.
  +RUN gunzip /usr/share/doc/libnccl2/NCCL-SLA.txt.gz && \
  + cp /usr/share/doc/libnccl2/NCCL-SLA.txt /usr/local/cuda/
  +
  RUN curl -fSsL -O https://bootstrap.pypa.io/get-pip.py && \
  python get-pip.py && \
  rm get-pip.py
  @@ -93,10 +102,13 @@ RUN git clone --branch=r1.9 --depth=1 https://github.com/tensorflow/tensorflow.g
  ENV CI_BUILD_PYTHON python
  ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
  ENV TF_NEED_CUDA 1
  -ENV TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.0,6.1
  +ENV TF_CUDA_COMPUTE_CAPABILITIES=3.5,5.2,6.0,6.1,7.0
  ENV TF_CUDA_VERSION=9.0
  ENV TF_CUDNN_VERSION=7
  +# NCCL 2.x
  +ENV TF_NCCL_VERSION=2
  +
  RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \
  LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:${LD_LIBRARY_PATH} \
  tensorflow/tools/ci_build/builds/configured GPU \
  