FROM nvidia/cuda:7.5-cudnn5-runtime-ubuntu14.04 MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com> # ENV variables ARG WITH_AVX ARG WITH_DOC ARG WITH_STYLE_CHECK ENV WITH_GPU=OFF ENV WITH_AVX=${WITH_AVX:-ON} ENV WITH_DOC=${WITH_DOC:-OFF} ENV WITH_STYLE_CHECK=${WITH_STYLE_CHECK:-OFF} ENV HOME /root ENV LANG en_US.UTF-8 # Use Fix locales to en_US.UTF-8 RUN sed 's@http:\/\/archive.ubuntu.com\/ubuntu\/@mirror:\/\/mirrors.ubuntu.com\/mirrors.txt@' -i /etc/apt/sources.list && \ apt-get update && \ apt-get install -y python python-pip libgfortran3 && \ apt-get clean -y && \ pip install --upgrade pip && \ pip install -U 'protobuf==3.1.0' RUN pip install numpy # Use different deb file when building different type of images ADD dist/cpu-noavx/*.deb /usr/local/opt/paddle/deb/cpu-noavx/ RUN dpkg --force-all -i /usr/local/opt/paddle/deb/cpu-noavx/*.deb && rm -f /usr/local/opt/paddle/deb/cpu-noavx/*.deb ENV PATH="/usr/local/opt/paddle/bin/:${PATH}" # default command shows the paddle version and exit CMD ["paddle", "version"]
