Dockerfile.gpu 1.3 KB
Newer Older
Y
Yi Wang 已提交
1 2
FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04
MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>
3

4 5
RUN apt-get update \
    && apt-get install -y cmake libprotobuf-dev protobuf-compiler git \
6 7 8
    libgoogle-glog-dev libgflags-dev libatlas-dev libatlas3-base g++ m4 python-pip \
    python-protobuf python-numpy python-dev swig openssh-server \
    wget unzip python-matplotlib tar xz-utils bzip2 gzip coreutils \
9 10 11
    sed grep graphviz libjpeg-dev zlib1g-dev doxygen \
    clang-3.8 llvm-3.8 libclang-3.8-dev \
    && apt-get clean -y
12 13 14
RUN pip install BeautifulSoup docopt PyYAML pillow \
    'sphinx>=1.4.0' sphinx_rtd_theme breathe recommonmark

Y
Yi Wang 已提交
15 16 17
ARG WITH_AVX
ENV WITH_AVX=${WITH_AVX:-ON}
ENV WITH_GPU=ON
18 19 20

RUN mkdir /paddle
COPY . /paddle/
Y
Yi Wang 已提交
21
RUN /paddle/paddle/scripts/docker/build.sh
22
VOLUME ["/usr/share/nginx/html/data", "/usr/share/nginx/html/paddle"]
23 24 25 26 27 28 29 30 31 32 33 34

RUN echo 'export LD_LIBRARY_PATH=/usr/lib64:${LD_LIBRARY_PATH}' >> /etc/profile
RUN pip install /usr/local/opt/paddle/share/wheels/*.whl
RUN paddle version  # print version after build

# 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
CMD ["/usr/sbin/sshd", "-D"]