From 64be2ecc07c698df05d88051ec42a0409d1a9863 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 28 Sep 2018 13:36:18 -0700 Subject: Do not pass in the bazel default toolchain via extra_toolchains. Without this the default toolchain is used for a subset of the build and the tests do not actually run on GPUs. This uncovered a setup problem in the Docker image that needed fixing. PiperOrigin-RevId: 214987676 --- tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/tools') diff --git a/tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 b/tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 index a30858db82..dd8d705331 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 +++ b/tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 @@ -26,7 +26,7 @@ ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0" ENV NCCL_VERSION 2.2.13 -ENV CUDNN_VERSION 7.2.1.38 +ENV CUDNN_VERSION 7.1.4.18 # TODO(b/110903506): /usr/loca/cuda/lib64/stubs should not be needed in # LD_LIBRARY_PATH. The stubs/libcuda.so is not meant to used at runtime. The -- cgit v1.2.3