@@ -4,7 +4,7 @@ function make_ubuntu_dockerfile(){
44 sed ' s/<baseimg>/10.1-cudnn7-devel-ubuntu16.04/g' ./Dockerfile.ubuntu > ${dockerfile_name}
55 sed -i ' s#liblzma-dev#liblzma-dev openmpi-bin openmpi-doc libopenmpi-dev#g' ${dockerfile_name}
66 dockerfile_line=` wc -l ${dockerfile_name} | awk ' {print $1}' `
7- sed -i " ${dockerfile_line} i RUN wget --no-check-certificate -q https://paddle-edl.bj.bcebos.com/hadoop-2.7.7.tar.gz && \
7+ sed -i " ${dockerfile_line} i RUN wget --no-check-certificate -q https://paddle-edl.bj.bcebos.com/hadoop-2.7.7.tar.gz && \
88 tar -xzf hadoop-2.7.7.tar.gz && mv hadoop-2.7.7 /usr/local/" ${dockerfile_name}
99 sed -i ' s#<install_gcc>#WORKDIR /usr/bin \
1010 COPY tools/dockerfile/build_scripts /build_scripts \
@@ -15,7 +15,9 @@ function make_ubuntu_dockerfile(){
1515 RUN ln -s /usr/local/gcc-8.2/bin/gcc /usr/bin/gcc \
1616 RUN ln -s /usr/local/gcc-8.2/bin/g++ /usr/bin/g++ \
1717 ENV PATH=/usr/local/gcc-8.2/bin:$PATH #g' ${dockerfile_name}
18-
18+ sed -i ' s#bash /build_scripts/install_nccl2.sh#wget --no-proxy https://nccl2-deb.cdn.bcebos.com/nccl-repo-ubuntu1604-2.7.8-ga-cuda10.1_1-1_amd64.deb && \
19+ dpkg -i nccl-repo-ubuntu1604-2.7.8-ga-cuda10.1_1-1_amd64.deb && \
20+ apt-get install -y libnccl2=2.7.8-1+cuda10.1 libnccl-dev=2.7.8-1+cuda10.1 --allow-change-held-packages #g' ${dockerfile_name}
1921}
2022
2123
@@ -27,8 +29,15 @@ function make_centos_dockerfile(){
2729 sed -i " ${dockerfile_line} i RUN ln -s /usr/lib64/libz.so /usr/local/lib/libz.so && \
2830 ln -s /usr/local/lib/libnccl.so /usr/local/cuda/lib64/ && \
2931 rm -rf /usr/include/NvInfer*" ${dockerfile_name}
30- sed -i " ${dockerfile_line} i RUN wget --no-check-certificate -q https://paddle-edl.bj.bcebos.com/hadoop-2.7.7.tar.gz && \
32+ sed -i " ${dockerfile_line} i RUN wget --no-check-certificate -q https://paddle-edl.bj.bcebos.com/hadoop-2.7.7.tar.gz && \
3133 tar -xzf hadoop-2.7.7.tar.gz && mv hadoop-2.7.7 /usr/local/" ${dockerfile_name}
34+ sed -i ' s#RUN bash build_scripts/install_nccl2.sh##g' ${dockerfile_name}
35+ sed -i " ${dockerfile_line} i RUN wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-2.7.8-1+cuda10.2.x86_64.rpm && \
36+ wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-devel-2.7.8-1+cuda10.2.x86_64.rpm && \
37+ wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-static-2.7.8-1+cuda10.2.x86_64.rpm && \
38+ rpm -ivh libnccl-2.7.8-1+cuda10.2.x86_64.rpm && \
39+ rpm -ivh libnccl-devel-2.7.8-1+cuda10.2.x86_64.rpm && \
40+ rpm -ivh libnccl-static-2.7.8-1+cuda10.2.x86_64.rpm && rm -f /usr/local/include/nccl.h " ${dockerfile_name}
3241 sed -i ' s#<install_gcc>#WORKDIR /usr/bin \
3342 COPY tools/dockerfile/build_scripts /build_scripts \
3443 RUN bash /build_scripts/install_gcc.sh gcc82 \&\& rm -rf /build_scripts \
0 commit comments