未验证 提交 e576f266 编写于 作者: C chengduo 提交者: GitHub

update docker build (#18523)

test=develop
上级 832d8191
...@@ -824,7 +824,7 @@ EOF ...@@ -824,7 +824,7 @@ EOF
EOF EOF
if [[ ${WITH_GPU} == "ON" ]]; then if [[ ${WITH_GPU} == "ON" ]]; then
NCCL_DEPS="apt-get install -y --allow-downgrades libnccl2=2.2.13-1+cuda${CUDA_MAJOR} libnccl-dev=2.2.13-1+cuda${CUDA_MAJOR} || true" NCCL_DEPS="apt-get install -y --allow-change-held-packages libnccl2=2.4.7-1+cuda${CUDA_MAJOR} libnccl-dev=2.4.7-1+cuda${CUDA_MAJOR} || true"
else else
NCCL_DEPS="true" NCCL_DEPS="true"
fi fi
...@@ -915,12 +915,6 @@ EOF ...@@ -915,12 +915,6 @@ 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.
先完成此消息的编辑!
想要评论请 注册