From 6737c75883bb5d0c24578870ecbe08c1a65edd07 Mon Sep 17 00:00:00 2001
From: Helin Wang <helinwang@baidu.com>
Date: Tue, 21 Mar 2017 19:06:09 -0700
Subject: [PATCH] fix gpu dockerbuild

---
 paddle/scripts/docker/build.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh
index 285ebdb9c3..0b45668bb8 100755
--- a/paddle/scripts/docker/build.sh
+++ b/paddle/scripts/docker/build.sh
@@ -14,13 +14,16 @@ fi
 DOCKERFILE_GPU_ENV=""
 if [[ ${WITH_GPU:-OFF} == 'ON' ]]; then
     DOCKERFILE_GPU_ENV="ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:${LD_LIBRARY_PATH}"
+
+    # for cmake to find cudnn
+    ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so /usr/lib/libcudnn.so
 fi
 
 mkdir -p /paddle/build
 cd /paddle/build
 
 # build script will not fail if *.deb does not exist
-rm *.deb || true
+rm *.deb 2>/dev/null || true
 
 cmake .. \
       -DCMAKE_BUILD_TYPE=Release \