From f72ced8814bfca980665f8ea80bd2dd4a4e6e71c Mon Sep 17 00:00:00 2001
From: Jiabin Yang <marsyang199376@gmail.com>
Date: Wed, 3 Jul 2019 17:55:25 +0800
Subject: [PATCH] test=develop, fix docker with paddle nccl problem (#18451)

---
 paddle/scripts/paddle_build.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
index f4930c2bbdf..b6e689f9853 100755
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -921,6 +921,12 @@ EOF
     # default command shows the paddle version and exit
     CMD [${CMD}]
 EOF
+    if [[ ${WITH_GPU} == "ON"  ]]; then
+        cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF
+    RUN rm /usr/lib/x86_64-linux-gnu/libnccl.so
+    RUN ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/lib/x86_64-linux-gnu/libnccl.so
+EOF
+    fi
 }
 
 function gen_fluid_lib() {
-- 
GitLab