diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index 2f5fef36c423736666695c07ebf69d812c3488ed..595af0c29a5c2c30427a82205d3d56c49492c0c8 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -614,7 +614,24 @@ EOF CMD='"true"' fi - cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF + if [ "$1" == "cp35-cp35m" ]; then + cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF + ADD python/dist/*.whl / + # run paddle version to install python packages first + RUN apt-get update && ${NCCL_DEPS} + RUN apt-get install -y wget python3 python3-pip libgtk2.0-dev dmidecode python3-tk && \ + pip3 install opencv-python && pip3 install /*.whl; apt-get install -f -y && \ + apt-get clean -y && \ + rm -f /*.whl && \ + ${PADDLE_VERSION} && \ + ldconfig + ${DOCKERFILE_CUDNN_DSO} + ${DOCKERFILE_CUBLAS_DSO} + ${DOCKERFILE_GPU_ENV} + ENV NCCL_LAUNCH_MODE PARALLEL +EOF + else + cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF ADD python/dist/*.whl / # run paddle version to install python packages first RUN apt-get update && ${NCCL_DEPS} @@ -629,6 +646,8 @@ EOF ${DOCKERFILE_GPU_ENV} ENV NCCL_LAUNCH_MODE PARALLEL EOF + fi + if [[ ${WITH_GOLANG:-OFF} == "ON" ]]; then cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF ADD go/cmd/pserver/pserver /usr/bin/ @@ -703,7 +722,7 @@ function main() { build) cmake_gen ${PYTHON_ABI:-""} build - gen_dockerfile + gen_dockerfile ${PYTHON_ABI:-""} ;; build_android) build_android @@ -730,7 +749,7 @@ function main() { gen_html ;; dockerfile) - gen_dockerfile + gen_dockerfile ${PYTHON_ABI:-""} ;; capi) cmake_gen ${PYTHON_ABI:-""}