Dockerfile.cuda10.1-cudnn7.devel 6.9 KB
Newer Older
W
wangjiawei04 已提交
1 2 3
# A image for building paddle binaries
# Use cuda devel base image for both cpu and gpu environment
# When you modify it, please be aware of cudnn-runtime version
W
wangjiawei04 已提交
4
FROM hub.baidubce.com/ctr/cuda:10.1-cudnn7-devel-ubuntu16.04
W
wangjiawei04 已提交
5 6 7 8 9 10 11 12 13 14 15
MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>

# ENV variables
ARG WITH_GPU
ARG WITH_AVX

ENV WITH_GPU=${WITH_GPU:-ON}
ENV WITH_AVX=${WITH_AVX:-ON}

ENV HOME /root
# Add bash enhancements
W
wangjiawei04 已提交
16
COPY tools/dockerfiles/root/ /root/
W
wangjiawei04 已提交
17 18 19 20 21 22 23

# Prepare packages for Python
RUN apt-get update && \
    apt-get install -y make build-essential libssl-dev zlib1g-dev libbz2-dev \
    libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev libncursesw5-dev \
    xz-utils tk-dev libffi-dev liblzma-dev

W
wangjiawei04 已提交
24 25 26 27 28
RUN apt-get update && \
    apt-get install -y --allow-downgrades --allow-change-held-packages \
    patchelf git python-pip python-dev python-opencv openssh-server bison \
    wget unzip unrar tar xz-utils bzip2 gzip coreutils ntp \
    curl sed grep graphviz libjpeg-dev zlib1g-dev  \
W
wangjiawei04 已提交
29
    python-matplotlib unzip \
W
wangjiawei04 已提交
30 31 32 33 34 35 36 37 38 39 40 41
    automake locales clang-format swig  \
    liblapack-dev liblapacke-dev libcurl4-openssl-dev \
    net-tools libtool module-init-tools vim && \
    apt-get clean -y

RUN ln -s /usr/lib/x86_64-linux-gnu/libssl.so /usr/lib/libssl.so.10 && \
    ln -s /usr/lib/x86_64-linux-gnu/libcrypto.so /usr/lib/libcrypto.so.10

RUN wget https://github.com/koalaman/shellcheck/releases/download/v0.7.1/shellcheck-v0.7.1.linux.x86_64.tar.xz -O shellcheck-v0.7.1.linux.x86_64.tar.xz && \
    tar -xf shellcheck-v0.7.1.linux.x86_64.tar.xz && cp  shellcheck-v0.7.1/shellcheck /usr/bin/shellcheck && \
    rm -rf shellcheck-v0.7.1.linux.x86_64.tar.xz shellcheck-v0.7.1

W
wangjiawei04 已提交
42
# Downgrade gcc&&g++
W
wangjiawei04 已提交
43
WORKDIR /usr/bin 
W
wangjiawei04 已提交
44
      COPY tools/dockerfiles/build_scripts /build_scripts 
W
wangjiawei04 已提交
45 46 47 48 49 50 51 52 53 54 55 56
      RUN bash /build_scripts/install_gcc.sh gcc82 && rm -rf /build_scripts 
      RUN cp gcc gcc.bak && cp g++ g++.bak && rm gcc && rm g++ 
      RUN ln -s /usr/local/gcc-8.2/bin/gcc /usr/local/bin/gcc 
      RUN ln -s /usr/local/gcc-8.2/bin/g++ /usr/local/bin/g++ 
      RUN ln -s /usr/local/gcc-8.2/bin/gcc /usr/bin/gcc 
      RUN ln -s /usr/local/gcc-8.2/bin/g++ /usr/bin/g++ 
      ENV PATH=/usr/local/gcc-8.2/bin:$PATH 

# install cmake
WORKDIR /home
RUN wget -q https://cmake.org/files/v3.16/cmake-3.16.0-Linux-x86_64.tar.gz && tar -zxvf cmake-3.16.0-Linux-x86_64.tar.gz && rm cmake-3.16.0-Linux-x86_64.tar.gz
ENV PATH=/home/cmake-3.16.0-Linux-x86_64/bin:$PATH
W
wangjiawei04 已提交
57 58 59 60

# Install Python3.6
RUN mkdir -p /root/python_build/ && wget -q https://www.sqlite.org/2018/sqlite-autoconf-3250300.tar.gz && \
    tar -zxf sqlite-autoconf-3250300.tar.gz && cd sqlite-autoconf-3250300 && \
W
wangjiawei04 已提交
61 62 63
    ./configure -prefix=/usr/local && make -j8 && make install && cd ../ && rm sqlite-autoconf-3250300.tar.gz

RUN wget -q https://www.python.org/ftp/python/3.6.0/Python-3.6.0.tgz && \
W
wangjiawei04 已提交
64 65
    tar -xzf Python-3.6.0.tgz && cd Python-3.6.0 && \
    CFLAGS="-Wformat" ./configure --prefix=/usr/local/ --enable-shared > /dev/null && \
W
wangjiawei04 已提交
66
    make -j8 > /dev/null && make altinstall > /dev/null && ldconfig && cd .. && rm -rf Python-3.6.0*
W
wangjiawei04 已提交
67 68 69 70 71

# Install Python3.7
RUN wget -q https://www.python.org/ftp/python/3.7.0/Python-3.7.0.tgz && \
    tar -xzf Python-3.7.0.tgz && cd Python-3.7.0 && \
    CFLAGS="-Wformat" ./configure --prefix=/usr/local/ --enable-shared > /dev/null && \
W
wangjiawei04 已提交
72
    make -j8 > /dev/null && make altinstall > /dev/null && ldconfig && cd .. && rm -rf Python-3.7.0*
W
wangjiawei04 已提交
73

W
wangjiawei04 已提交
74 75 76 77
# Install Python3.8
RUN wget -q https://www.python.org/ftp/python/3.8.0/Python-3.8.0.tgz && \
    tar -xzf Python-3.8.0.tgz && cd Python-3.8.0 && \
    CFLAGS="-Wformat" ./configure --prefix=/usr/local/ --enable-shared > /dev/null && \
W
wangjiawei04 已提交
78
    make -j8 > /dev/null && make altinstall > /dev/null && ldconfig && cd .. && rm -rf Python-3.8.0*
W
wangjiawei04 已提交
79

W
wangjiawei04 已提交
80 81
ENV LD_LIBRARY_PATH=/usr/local/lib:${LD_LIBRARY_PATH}
RUN ln -sf /usr/local/bin/python3.6 /usr/local/bin/python3 && ln -sf /usr/local/bin/python3.6 /usr/bin/python3 && ln -sf /usr/local/bin/pip3.6 /usr/local/bin/pip3 && ln -sf /usr/local/bin/pip3.6 /usr/bin/pip3
W
wangjiawei04 已提交
82

W
wangjiawei04 已提交
83
RUN rm -r /root/python_build
W
wangjiawei04 已提交
84 85

# Install Go and glide
W
wangjiawei04 已提交
86
RUN wget -qO- https://dl.google.com/go/go1.14.linux-amd64.tar.gz | \
W
wangjiawei04 已提交
87
    tar -xz -C /usr/local && \
W
wangjiawei04 已提交
88 89 90 91 92 93 94
    mkdir /root/go && \
    mkdir /root/go/bin && \
    mkdir /root/go/src && \
    echo "GOROOT=/usr/local/go" >> /root/.bashrc && \
    echo "GOPATH=/root/go" >> /root/.bashrc && \
    echo "PATH=/usr/local/go/bin:/root/go/bin:$PATH" >> /root/.bashrc
ENV GOROOT=/usr/local/go GOPATH=/root/go
W
wangjiawei04 已提交
95
# should not be in the same line with GOROOT definition, otherwise docker build could not find GOROOT.
W
wangjiawei04 已提交
96
ENV PATH=usr/local/go/bin:/root/go/bin:${PATH}
W
wangjiawei04 已提交
97 98 99 100 101 102 103 104

# Install TensorRT
# following TensorRT.tar.gz is not the default official one, we do two miny changes:
# 1. Remove the unnecessary files to make the library small. TensorRT.tar.gz only contains include and lib now,
#    and its size is only one-third of the official one.
# 2. Manually add ~IPluginFactory() in IPluginFactory class of NvInfer.h, otherwise, it couldn't work in paddle.
#    See https://github.com/PaddlePaddle/Paddle/issues/10129 for details.

W
wangjiawei04 已提交
105
# Downgrade TensorRT 
W
wangjiawei04 已提交
106
COPY tools/dockerfiles/build_scripts /build_scripts
B
fix trt  
bjjwwang 已提交
107
RUN bash /build_scripts/install_trt.sh cuda10.1  
W
wangjiawei04 已提交
108
RUN rm -rf /build_scripts
W
wangjiawei04 已提交
109 110 111 112 113 114 115

# git credential to skip password typing
RUN git config --global credential.helper store

# Fix locales to en_US.UTF-8
RUN localedef -i en_US -f UTF-8 en_US.UTF-8

W
wangjiawei04 已提交
116
RUN apt-get install libprotobuf-dev -y
W
wangjiawei04 已提交
117 118 119 120

# Older versions of patchelf limited the size of the files being processed and were fixed in this pr.
# https://github.com/NixOS/patchelf/commit/ba2695a8110abbc8cc6baf0eea819922ee5007fa
# So install a newer version here.
W
wangjiawei04 已提交
121
RUN wget -q https://paddle-ci.cdn.bcebos.com/patchelf_0.10-2_amd64.deb && \
W
wangjiawei04 已提交
122 123 124
    dpkg -i patchelf_0.10-2_amd64.deb

# Configure OpenSSH server. c.f. https://docs.docker.com/engine/examples/running_ssh_service
W
wangjiawei04 已提交
125
RUN mkdir /var/run/sshd && echo 'root:root' | chpasswd && sed -ri 's/^PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config && sed -ri 's/UsePAM yes/#UsePAM yes/g' /etc/ssh/sshd_config
W
wangjiawei04 已提交
126
CMD source ~/.bashrc
W
wangjiawei04 已提交
127 128 129 130 131 132 133 134 135 136

# ccache 3.7.9
RUN wget https://paddle-ci.gz.bcebos.com/ccache-3.7.9.tar.gz && \
    tar xf ccache-3.7.9.tar.gz && mkdir /usr/local/ccache-3.7.9 && cd ccache-3.7.9 && \
    ./configure -prefix=/usr/local/ccache-3.7.9 && \
    make -j8 && make install && \
    ln -s /usr/local/ccache-3.7.9/bin/ccache /usr/local/bin/ccache

RUN python3.8 -m pip install --upgrade pip requests && \
    python3.7 -m pip install --upgrade pip requests && \
W
wangjiawei04 已提交
137
    python3.6 -m pip install --upgrade pip requests 
W
wangjiawei04 已提交
138 139 140 141 142 143 144 145 146

RUN wget https://paddle-serving.bj.bcebos.com/others/centos_ssl.tar && \
    tar xf centos_ssl.tar && rm -rf centos_ssl.tar && \
    mv libcrypto.so.1.0.2k /usr/lib/libcrypto.so.1.0.2k && mv libssl.so.1.0.2k /usr/lib/libssl.so.1.0.2k && \
    ln -sf /usr/lib/libcrypto.so.1.0.2k /usr/lib/libcrypto.so.10 && \
    ln -sf /usr/lib/libssl.so.1.0.2k /usr/lib/libssl.so.10 && \
    ln -sf /usr/lib/libcrypto.so.10 /usr/lib/libcrypto.so && \
    ln -sf /usr/lib/libssl.so.10 /usr/lib/libssl.so

W
wangjiawei04 已提交
147
EXPOSE 22