|
|
|
@ -824,7 +824,7 @@ EOF
|
|
|
|
|
EOF
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
NCCL_DEPS="true"
|
|
|
|
|
fi
|
|
|
|
@ -915,12 +915,6 @@ 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() {
|
|
|
|
|