diff --git a/Dockerfile b/Dockerfile index 9459552890fac393304d25c19658a0480c1909b2..84e1edbee91b0c3877ec7c55161af8c3bdf1c0dc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -43,6 +43,8 @@ RUN wget -q https://www.python.org/ftp/python/3.7.0/Python-3.7.0.tgz && \ CFLAGS="-Wformat" ./configure --prefix=/usr/local/ --enable-shared > /dev/null && \ make -j8 > /dev/null && make altinstall > /dev/null +RUN rm -r /root/python_build + RUN apt-get update && \ apt-get install -y --allow-downgrades patchelf \ python3 python3-dev python3-pip \ diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index 86925b26e7e894d7dba1feac68b79e66316752d7..c4e283e76a54b9883729b32381911b5a3fe5f658 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -671,6 +671,55 @@ EOF ${DOCKERFILE_CUBLAS_DSO} ${DOCKERFILE_GPU_ENV} ENV NCCL_LAUNCH_MODE PARALLEL +EOF + elif [ "$1" == "cp36-cp36m" ]; then + cat >> ${PADDLE_ROOT}/build/Dockerfile < /dev/null && \ + make -j8 > /dev/null && make altinstall > /dev/null + RUN apt-get install -y libgtk2.0-dev dmidecode python3-tk && \ + pip3.6 install opencv-python && pip3.6 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 + elif [ "$1" == "cp37-cp37m" ]; then + cat >> ${PADDLE_ROOT}/build/Dockerfile < /dev/null && \ + make -j8 > /dev/null && make altinstall > /dev/null + RUN apt-get install -y libgtk2.0-dev dmidecode python3-tk && \ + pip3.7 install opencv-python && pip3.7 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 <