diff options
author | 2018-07-20 17:07:22 -0700 | |
---|---|---|
committer | 2018-07-20 17:07:28 -0700 | |
commit | 9efb87bfc40296c7804ba8ad1a0af830fc548be9 (patch) | |
tree | 1c36d0319328dfa5f4ed10607b0851f8f56048e0 /tensorflow/tools/ci_build/Dockerfile.gpu | |
parent | 9dbbea7a4327cd694ccff1f9edf6ae1af4329362 (diff) | |
parent | c0d998e3a772b0021b6283145e2aa24b701a0e7f (diff) |
Merge pull request #20981 from case540:nccl_configure_cherrypick
PiperOrigin-RevId: 205475673
Diffstat (limited to 'tensorflow/tools/ci_build/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index 6ab703d93d..33d0425918 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -27,6 +27,11 @@ RUN /install/install_golang.sh COPY install/.bazelrc /etc/bazel.bazelrc ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH +# 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 + # Configure the build for our CUDA configuration. ENV TF_NEED_CUDA 1 ENV TF_CUDA_COMPUTE_CAPABILITIES 3.0 |