diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index 9632eaec005df6cdfd289f0e4e6879ad2168cdeb..e69492da906bd58be3cd70e0502cf3f8f0cee3f2 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 <