From d7dbf21bda3fae90ced99db1bb5c592264c02526 Mon Sep 17 00:00:00 2001 From: Austin Anderson Date: Tue, 12 Dec 2017 21:24:11 -0800 Subject: [PATCH] Explicitly specify CUDA and CUDNN versions (#15328) A workaround to the base problem causing https://github.com/tensorflow/tensorflow/issues/15291 --- tensorflow/tools/docker/Dockerfile.devel-gpu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 339a39bcf3e..9f4cc74a66f 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -81,7 +81,8 @@ 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_VERSION=9.0 +ENV TF_CUDNN_VERSION=7 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} \