Dockerfile 4.6 KB
Newer Older
Y
yi.wu 已提交
1 2
# A image for building paddle binaries
# Use cuda devel base image for both cpu and gpu environment
D
dzhwinter 已提交
3
# When you modify it, please be aware of cudnn-runtime version
G
gongweibao 已提交
4
# and libcudnn.so.x in paddle/scripts/docker/build.sh
Y
init  
Yang Yang 已提交
5
FROM nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04
Y
Yi Wang 已提交
6
MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>
7

Y
yangwenbo02 已提交
8
ARG UBUNTU_MIRROR
Y
yi.wu 已提交
9
RUN /bin/bash -c 'if [[ -n ${UBUNTU_MIRROR} ]]; then sed -i 's#http://archive.ubuntu.com/ubuntu#${UBUNTU_MIRROR}#g' /etc/apt/sources.list; fi'
L
liaogang 已提交
10

L
liaogang 已提交
11
# ENV variables
Y
yi.wu 已提交
12
ARG WITH_GPU
L
liaogang 已提交
13 14 15
ARG WITH_AVX
ARG WITH_DOC

Y
update  
Yancey1989 已提交
16
ENV WOBOQ OFF
17
ENV WITH_GPU=${WITH_GPU:-ON}
L
liaogang 已提交
18 19 20 21 22 23 24
ENV WITH_AVX=${WITH_AVX:-ON}
ENV WITH_DOC=${WITH_DOC:-OFF}

ENV HOME /root
# Add bash enhancements
COPY ./paddle/scripts/docker/root/ /root/

M
minqiyang 已提交
25 26 27 28
RUN apt-get update || \
    apt-get install -y --allow-unauthenticated libnccl2=2.1.2-1+cuda8.0 libnccl-dev=2.1.2-1+cuda8.0

RUN apt-get install -y --allow-downgrades patchelf \
29
    git python-pip python-dev python-opencv openssh-server bison \
M
minqiyang 已提交
30
    python3 python3-pip python3-dev \
D
dongzhihong 已提交
31
    wget unzip unrar tar xz-utils bzip2 gzip coreutils ntp \
Y
Yi Wang 已提交
32
    curl sed grep graphviz libjpeg-dev zlib1g-dev  \
Y
Yu Yang 已提交
33
    python-matplotlib gcc-4.8 g++-4.8 \
Y
yuyang18 已提交
34
    automake locales clang-format swig cmake  \
35
    liblapack-dev liblapacke-dev \
Q
qijun 已提交
36
    clang-3.8 llvm-3.8 libclang-3.8-dev \
37
    net-tools libtool ccache && \
Q
qijun 已提交
38 39
    apt-get clean -y

W
wuyi05 已提交
40
# Install Go and glide
Y
Yi Wang 已提交
41 42
RUN wget -qO- https://storage.googleapis.com/golang/go1.8.1.linux-amd64.tar.gz | \
    tar -xz -C /usr/local && \
H
Helin Wang 已提交
43
    mkdir /root/gopath && \
W
wuyi05 已提交
44
    mkdir /root/gopath/bin && \
Y
Yi Wang 已提交
45
    mkdir /root/gopath/src
H
Helin Wang 已提交
46 47
ENV GOROOT=/usr/local/go GOPATH=/root/gopath
# should not be in the same line with GOROOT definition, otherwise docker build could not find GOROOT.
W
wuyi05 已提交
48
ENV PATH=${PATH}:${GOROOT}/bin:${GOPATH}/bin
W
wuyi05 已提交
49
# install glide
50
RUN curl -s -q https://glide.sh/get | sh
H
Helin Wang 已提交
51

Y
Yan Chunwei 已提交
52
# Install TensorRT
L
Luo Tao 已提交
53 54 55 56 57
# 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.
G
guochaorong 已提交
58
RUN wget -qO- http://paddlepaddledeps.cdn.bcebos.com/TensorRT-4.0.0.3.Ubuntu-16.04.4.x86_64-gnu.cuda-8.0.cudnn7.0.tar.gz | \
Y
Yan Chunwei 已提交
59 60 61 62
    tar -xz -C /usr/local && \
    cp -rf /usr/local/TensorRT/include /usr && \
    cp -rf /usr/local/TensorRT/lib /usr

L
liaogang 已提交
63 64 65 66 67 68
# 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

Y
yi.wu 已提交
69 70
# FIXME: due to temporary ipykernel dependency issue, specify ipykernel jupyter
# version util jupyter fixes this issue.
_青葱's avatar
_青葱 已提交
71 72 73 74

# specify sphinx version as 1.5.6 and remove -U option for [pip install -U
# sphinx-rtd-theme] since -U option will cause sphinx being updated to newest
# version(1.7.1 for now), which causes building documentation failed.
75
RUN easy_install -U pip && \
D
dong zhihong 已提交
76
    pip install -U wheel && \
77
    pip install -U docopt PyYAML sphinx==1.5.6 && \
M
minqiyang 已提交
78 79 80 81 82
    pip install sphinx-rtd-theme==0.1.9 recommonmark && \
    pip3 install --upgrade pip && \
    pip3 install -U wheel && \
    pip3 install -U docopt PyYAML sphinx==1.5.6 && \
    pip3 install sphinx-rtd-theme==0.1.9 recommonmark
D
dong zhihong 已提交
83 84

RUN pip install pre-commit 'ipython==5.3.0' && \
W
wuyi05 已提交
85
    pip install 'ipykernel==4.6.0' 'jupyter==1.0.0' && \
M
minqiyang 已提交
86 87 88 89
    pip install opencv-python && \
    pip3 install pre-commit 'ipython==5.3.0' && \
    pip3 install 'ipykernel==4.6.0' 'jupyter==1.0.0' && \
    pip3 install opencv-python
D
dong zhihong 已提交
90

G
gongweibao 已提交
91
#For docstring checker
92
RUN pip install pylint pytest astroid isort LinkChecker
M
minqiyang 已提交
93
RUN pip3 install pylint pytest astroid isort
G
gongweibao 已提交
94

D
dong zhihong 已提交
95 96
COPY ./python/requirements.txt /root/
RUN pip install -r /root/requirements.txt
M
minqiyang 已提交
97
RUN pip3 install -r /root/requirements.txt
L
liaogang 已提交
98

Y
Yi Wang 已提交
99 100
# To fix https://github.com/PaddlePaddle/Paddle/issues/1954, we use
# the solution in https://urllib3.readthedocs.io/en/latest/user-guide.html#ssl-py2
Y
Yi Wang 已提交
101 102
RUN apt-get install -y libssl-dev libffi-dev
RUN pip install certifi urllib3[secure]
M
minqiyang 已提交
103
RUN pip3 install certifi urllib3[secure]
Y
Yi Wang 已提交
104

Q
qijun 已提交
105

Y
Yi Wang 已提交
106 107 108 109 110 111
# Install woboq_codebrowser to /woboq
RUN git clone https://github.com/woboq/woboq_codebrowser /woboq && \
    (cd /woboq \
     cmake -DLLVM_CONFIG_EXECUTABLE=/usr/bin/llvm-config-3.8 \
           -DCMAKE_BUILD_TYPE=Release . \
     make)
112 113 114 115 116 117 118

# Configure OpenSSH server. c.f. https://docs.docker.com/engine/examples/running_ssh_service
RUN mkdir /var/run/sshd
RUN echo 'root:root' | chpasswd
RUN sed -ri 's/^PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN sed -ri 's/UsePAM yes/#UsePAM yes/g' /etc/ssh/sshd_config
EXPOSE 22