test=develop, fix docker with paddle nccl problem (#18451)

6 years ago · f72ced8814
parent 3f3112ceb0
commit f72ced8814
1 changed files with 6 additions and 0 deletions
--- 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() {