|
|
@ -41,6 +41,7 @@ function make_centos_dockerfile(){
|
|
|
|
sed "s/<baseimg>/11.0-cudnn8-devel-centos7/g" Dockerfile.centos >${dockerfile_name}
|
|
|
|
sed "s/<baseimg>/11.0-cudnn8-devel-centos7/g" Dockerfile.centos >${dockerfile_name}
|
|
|
|
sed -i "s#COPY build_scripts /build_scripts#COPY tools/dockerfile/build_scripts ./build_scripts#g" ${dockerfile_name}
|
|
|
|
sed -i "s#COPY build_scripts /build_scripts#COPY tools/dockerfile/build_scripts ./build_scripts#g" ${dockerfile_name}
|
|
|
|
dockerfile_line=$(wc -l ${dockerfile_name}|awk '{print $1}')
|
|
|
|
dockerfile_line=$(wc -l ${dockerfile_name}|awk '{print $1}')
|
|
|
|
|
|
|
|
sed -i "${dockerfile_line}i RUN rm -f /usr/bin/cc && ln -s /usr/local/gcc-8.2/bin/gcc /usr/bin/cc" ${dockerfile_name}
|
|
|
|
sed -i "${dockerfile_line}i RUN ln -s /usr/lib64/libz.so /usr/local/lib/libz.so \\
|
|
|
|
sed -i "${dockerfile_line}i RUN ln -s /usr/lib64/libz.so /usr/local/lib/libz.so \\
|
|
|
|
RUN ln -s /usr/local/lib/libnccl.so /usr/local/cuda/lib64/ \\
|
|
|
|
RUN ln -s /usr/local/lib/libnccl.so /usr/local/cuda/lib64/ \\
|
|
|
|
RUN rm -rf /usr/include/NvInfer*" ${dockerfile_name}
|
|
|
|
RUN rm -rf /usr/include/NvInfer*" ${dockerfile_name}
|
|
|
|