diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh index 1779a0753beefe0aac6601de9b8dd62408953a8a..c44874eede03a8b1060b15e175ad89622f925940 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