From c0d998e3a772b0021b6283145e2aa24b701a0e7f Mon Sep 17 00:00:00 2001 From: Toby Boyd Date: Tue, 17 Jul 2018 14:26:44 -0700 Subject: Link lib and header where ./configure expects --- tensorflow/tools/ci_build/Dockerfile.gpu | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'tensorflow/tools/ci_build/Dockerfile.gpu') diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index 7591ecc04e..46538d9ec7 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -22,6 +22,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 -- cgit v1.2.3