File tree Expand file tree Collapse file tree 1 file changed +22
-3
lines changed Expand file tree Collapse file tree 1 file changed +22
-3
lines changed Original file line number Diff line number Diff line change 614
614
CMD=' "true"'
615
615
fi
616
616
617
- cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
617
+ if [ " $1 " == " cp35-cp35m" ]; then
618
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
619
+ ADD python/dist/*.whl /
620
+ # run paddle version to install python packages first
621
+ RUN apt-get update && ${NCCL_DEPS}
622
+ RUN apt-get install -y wget python3 python3-pip libgtk2.0-dev dmidecode python3-tk && \
623
+ pip3 install opencv-python && pip3 install /*.whl; apt-get install -f -y && \
624
+ apt-get clean -y && \
625
+ rm -f /*.whl && \
626
+ ${PADDLE_VERSION} && \
627
+ ldconfig
628
+ ${DOCKERFILE_CUDNN_DSO}
629
+ ${DOCKERFILE_CUBLAS_DSO}
630
+ ${DOCKERFILE_GPU_ENV}
631
+ ENV NCCL_LAUNCH_MODE PARALLEL
632
+ EOF
633
+ else
634
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
618
635
ADD python/dist/*.whl /
619
636
# run paddle version to install python packages first
620
637
RUN apt-get update && ${NCCL_DEPS}
629
646
${DOCKERFILE_GPU_ENV}
630
647
ENV NCCL_LAUNCH_MODE PARALLEL
631
648
EOF
649
+ fi
650
+
632
651
if [[ ${WITH_GOLANG:- OFF} == " ON" ]]; then
633
652
cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
634
653
ADD go/cmd/pserver/pserver /usr/bin/
@@ -703,7 +722,7 @@ function main() {
703
722
build)
704
723
cmake_gen ${PYTHON_ABI:- " " }
705
724
build
706
- gen_dockerfile
725
+ gen_dockerfile ${PYTHON_ABI :- " " }
707
726
;;
708
727
build_android)
709
728
build_android
@@ -730,7 +749,7 @@ function main() {
730
749
gen_html
731
750
;;
732
751
dockerfile)
733
- gen_dockerfile
752
+ gen_dockerfile ${PYTHON_ABI :- " " }
734
753
;;
735
754
capi)
736
755
cmake_gen ${PYTHON_ABI:- " " }
You can’t perform that action at this time.
0 commit comments