未验证 提交 f72ced88 编写于 作者: J Jiabin Yang 提交者: GitHub

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

上级 3f3112ce
...@@ -921,6 +921,12 @@ EOF ...@@ -921,6 +921,12 @@ EOF
# default command shows the paddle version and exit # default command shows the paddle version and exit
CMD [${CMD}] CMD [${CMD}]
EOF 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() { function gen_fluid_lib() {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册