Dockerfile 2.1 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
Y
Yi Wang 已提交
3 4
FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04
MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>
5

Y
yangwenbo02 已提交
6
ARG UBUNTU_MIRROR
Y
yi.wu 已提交
7
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 已提交
8

L
liaogang 已提交
9
# ENV variables
Y
update  
Yancey1989 已提交
10
ARG WOBOQ
Y
yi.wu 已提交
11
ARG WITH_GPU
L
liaogang 已提交
12 13 14 15
ARG WITH_AVX
ARG WITH_DOC
ARG WITH_STYLE_CHECK

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

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

L
liaogang 已提交
26
RUN apt-get update && \
L
liaogang 已提交
27
    apt-get install -y git python-pip python-dev openssh-server bison && \
L
liaogang 已提交
28 29
    apt-get install -y wget unzip tar xz-utils bzip2 gzip coreutils && \
    apt-get install -y curl sed grep graphviz libjpeg-dev zlib1g-dev && \
L
liaogang 已提交
30
    apt-get install -y python-numpy python-matplotlib gcc g++ gfortran && \
H
Helin Wang 已提交
31
    apt-get install -y automake locales clang-format-3.8 swig && \
L
liaogang 已提交
32 33
    apt-get clean -y

L
liaogang 已提交
34 35 36 37 38 39
# 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

40
RUN pip install --upgrade pip && \
L
liaogang 已提交
41
    pip install -U 'protobuf==3.1.0' && \
L
liaogang 已提交
42 43
    pip install -U wheel pillow BeautifulSoup && \
    pip install -U docopt PyYAML sphinx && \
Y
Yu Yang 已提交
44
    pip install -U sphinx-rtd-theme==0.1.9 recommonmark && \
L
liaogang 已提交
45
    pip install -U pre-commit 'requests==2.9.2' jupyter
L
liaogang 已提交
46 47

RUN curl -sSL https://cmake.org/files/v3.4/cmake-3.4.1.tar.gz | tar -xz && \
L
liaogang 已提交
48
    cd cmake-3.4.1 && ./bootstrap && make -j `nproc` && make install && \
L
liaogang 已提交
49
    cd .. && rm -rf cmake-3.4.1
50

Y
update  
Yancey1989 已提交
51
VOLUME ["/woboq_out"]
52 53 54 55 56 57 58

# 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
59

Y
yi.wu 已提交
60 61
# development image default do build work
CMD ["bash", "/paddle/paddle/scripts/docker/build.sh"]