gpu image dependencies

Author: yi.wu

paddle/scripts/docker/build.sh

@@ -14,6 +14,8 @@ mkdir -p /paddle/dist/gpu-noavx
 # Set BASE_IMAGE and DEB_PATH according to env variables
 if [ ${WITH_GPU} == "ON" ]; then
   BASE_IMAGE="nvidia/cuda:7.5-cudnn5-runtime-ubuntu14.04"
+  # additional packages to install when building gpu images
+  GPU_DOCKER_PKG="python-pip"
   if [ ${WITH_AVX} == "ON" ]; then
     DEB_PATH="dist/gpu/"
     DOCKER_SUFFIX="gpu"
@@ -125,7 +127,7 @@ ENV LANG en_US.UTF-8
 
 RUN ${MIRROR_UPDATE}
     apt-get update && \
-    apt-get install -y libgfortran3 && \
+    apt-get install -y libgfortran3 ${GPU_DOCKER_PKG} && \
     apt-get clean -y && \
     pip install --upgrade pip && \
     pip install -U 'protobuf==3.1.0' requests