diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-28 13:36:18 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 14:30:18 -0700 |
| commit | 64be2ecc07c698df05d88051ec42a0409d1a9863 (patch) | |
| tree | 0b96c238506c713bae3f6a99dc3399f268f61bb2 /tensorflow/tools | |
| parent | 19143aa0e2ac3cdf0d6826e7e1d00cd864080394 (diff) | |
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
Diffstat (limited to 'tensorflow/tools')
| -rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 | 2 |
1 files changed, 1 insertions, 1 deletions
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 |