From e8d33bee479a1bd574875f8b5ca1463955d6323d Mon Sep 17 00:00:00 2001
From: "yi.wu" <yi.wu@baifendian.com>
Date: Fri, 17 Mar 2017 21:25:19 +0800
Subject: [PATCH] gpu image dependencies

---
 paddle/scripts/docker/build.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh
index 1779a0753be..c44874eede0 100755
--- a/paddle/scripts/docker/build.sh
+++ b/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
-- 
GitLab