From 12ae354a106a858e3feb3a234d6ddd0fb6abf18c Mon Sep 17 00:00:00 2001 From: gongweibao Date: Sat, 21 Apr 2018 08:56:41 +0800 Subject: [PATCH] Fix cudnn version(#10088) --- Dockerfile | 3 +++ paddle/scripts/docker/build.sh | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 870304a6ac..9ac58f37f2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,8 @@ # A image for building paddle binaries # Use cuda devel base image for both cpu and gpu environment + +# When you modify it, please be aware of cudnn-runtime version +# and libcudnn.so.x in paddle/scripts/docker/build.sh FROM nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04 MAINTAINER PaddlePaddle Authors diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh index 86ef3e4df1..9462827022 100755 --- a/paddle/scripts/docker/build.sh +++ b/paddle/scripts/docker/build.sh @@ -155,7 +155,7 @@ EOF function gen_dockerfile() { # Set BASE_IMAGE according to env variables if [[ ${WITH_GPU} == "ON" ]]; then - BASE_IMAGE="nvidia/cuda:8.0-cudnn5-runtime-ubuntu16.04" + BASE_IMAGE="nvidia/cuda:8.0-cudnn7-runtime-ubuntu16.04" else BASE_IMAGE="ubuntu:16.04" fi @@ -164,7 +164,7 @@ function gen_dockerfile() { DOCKERFILE_CUDNN_DSO="" if [[ ${WITH_GPU:-OFF} == 'ON' ]]; then DOCKERFILE_GPU_ENV="ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:\${LD_LIBRARY_PATH}" - DOCKERFILE_CUDNN_DSO="RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.5 /usr/lib/x86_64-linux-gnu/libcudnn.so" + DOCKERFILE_CUDNN_DSO="RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7 /usr/lib/x86_64-linux-gnu/libcudnn.so" fi cat <