Refine Docker build

Author: reyoung

cmake/package.cmake

@@ -1,5 +1,4 @@
 set(CPACK_PACKAGE_NAME paddle)
-set(CPACK_PACKAGE_DESCRIPTION_SUMMARY "")
 set(CPACK_PACKAGE_VERSION_MAJOR ${PADDLE_MAJOR_VERSION})
 set(CPACK_PACKAGE_VERSION_MINOR ${PADDLE_MINOR_VERSION})
 set(CPACK_PACKAGE_VERSION_PATCH ${PADDLE_PATCH_VERSION})
@@ -12,6 +11,7 @@ set(CPACK_PACKAGE_DESCRIPTION_SUMMARY "Paddle")
 set(CPACK_PACKAGE_DESCRIPTION "")
 set(CPACK_DEBIAN_PACKAGE_DEPENDS "libpython2.7, libstdc++6, python-pip, curl, libgfortran3, python-pip-whl")
 set(CPACK_DEBIAN_PACKAGE_SECTION Devel)
+set(CPACK_DEBIAN_PACKAGE_VERSION ${PADDLE_VERSION})
 set(CPACK_DEBIAN_PACKAGE_CONTROL_EXTRA "${PROJ_ROOT}/paddle/scripts/deb/postinst")
 #set(CPACK_GENERATOR "DEB")
 # Start cpack

paddle/scripts/deb/postinst

@@ -0,0 +1,6 @@
+#!/bin/bash
+set -e
+echo "Post install paddle debian package."
+echo "Install some python package used for paddle. You can run "
+echo "  pip install /usr/opt/paddle/share/wheels/*.whl to install them."
+find /usr/ -name '*paddle*.whl' | xargs pip install

paddle/scripts/docker/build.sh

@@ -5,13 +5,8 @@ set -e
 # Set BASE_IMAGE according to env variables
 if [ ${WITH_GPU} == "ON" ]; then
   BASE_IMAGE="nvidia/cuda:8.0-cudnn5-runtime-ubuntu14.04"
-  # additional packages to install when building gpu images
-  GPU_DOCKER_PKG="python-pip python-dev"
 else
-  BASE_IMAGE="python:2.7.13-slim"
-  # FIXME: python base image uses different python version than WITH_GPU
-  # need to change PYTHONHOME to /usr/local when using python base image
-  CPU_DOCKER_PYTHON_HOME_ENV="ENV PYTHONHOME /usr/local"
+  BASE_IMAGE="ubuntu:14.04"
 fi
 
 DOCKERFILE_GPU_ENV=""
@@ -88,12 +83,6 @@ fi
 
 paddle version
 
-if [[ -n ${APT_MIRROR} ]]; then
-  MIRROR_UPDATE="sed -i '${APT_MIRROR}' /etc/apt/sources.list"
-else
-  MIRROR_UPDATE=""
-fi
-
 cat > /paddle/build/Dockerfile <<EOF
 FROM ${BASE_IMAGE}
 MAINTAINER PaddlePaddle Authors <[email protected]>
@@ -102,23 +91,13 @@ ENV LANG en_US.UTF-8
 # Use Fix locales to en_US.UTF-8
 EOF
 
-if [[ -n ${MIRROR_UPDATE} ]]; then
-cat >> /paddle/build/Dockerfile <<EOF
-RUN ${MIRROR_UPDATE}
-EOF
-fi
-
-if [[ -n ${GPU_DOCKER_PKG} ]]; then
+if [[ -n ${APT_MIRROR} ]]; then
 cat >> /paddle/build/Dockerfile <<EOF
-RUN apt-get update && \
-    apt-get install -y ${GPU_DOCKER_PKG} && \
-    apt-get clean -y
+RUN sed -i '${APT_MIRROR}' /etc/apt/sources.list
 EOF
 fi
 
 cat >> /paddle/build/Dockerfile <<EOF
-RUN pip install --upgrade pip
-
 # Use different deb file when building different type of images
 ADD build/*.deb /
 # run paddle version to install python packages first
@@ -127,7 +106,6 @@ RUN apt-get update &&\
     apt-get clean -y && \
     rm -f /*.deb && \
     paddle version
-${CPU_DOCKER_PYTHON_HOME_ENV}
 ${DOCKERFILE_CUDNN_DSO}
 ${DOCKERFILE_GPU_ENV}
 # default command shows the paddle version and exit

python/setup.py.in

@@ -16,7 +16,8 @@ setup(name='paddle',
       packages=packages,
       install_requires=[
           "numpy",
-          "protobuf==3.1.0"
+          "protobuf==3.1.0",
+          "matplotlib",
       ],
       package_dir={
           '': '${CMAKE_CURRENT_SOURCE_DIR}'