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 609
609
CMD=' "true"'
610
610
fi
611
611
612
- cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
612
+ if [ " $1 " == " cp35-cp35m" ]; then
613
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
614
+ ADD python/dist/*.whl /
615
+ # run paddle version to install python packages first
616
+ RUN apt-get update && ${NCCL_DEPS}
617
+ RUN apt-get install -y wget python3 python3-pip libgtk2.0-dev dmidecode python3-tk && \
618
+ pip3 install opencv-python && pip3 install /*.whl; apt-get install -f -y && \
619
+ apt-get clean -y && \
620
+ rm -f /*.whl && \
621
+ ${PADDLE_VERSION} && \
622
+ ldconfig
623
+ ${DOCKERFILE_CUDNN_DSO}
624
+ ${DOCKERFILE_CUBLAS_DSO}
625
+ ${DOCKERFILE_GPU_ENV}
626
+ ENV NCCL_LAUNCH_MODE PARALLEL
627
+ EOF
628
+ else
629
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
613
630
ADD python/dist/*.whl /
614
631
# run paddle version to install python packages first
615
632
RUN apt-get update && ${NCCL_DEPS}
624
641
${DOCKERFILE_GPU_ENV}
625
642
ENV NCCL_LAUNCH_MODE PARALLEL
626
643
EOF
644
+ fi
645
+
627
646
if [[ ${WITH_GOLANG:- OFF} == " ON" ]]; then
628
647
cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
629
648
ADD go/cmd/pserver/pserver /usr/bin/
@@ -698,7 +717,7 @@ function main() {
698
717
build)
699
718
cmake_gen ${PYTHON_ABI:- " " }
700
719
build
701
- gen_dockerfile
720
+ gen_dockerfile ${PYTHON_ABI :- " " }
702
721
;;
703
722
build_android)
704
723
build_android
@@ -725,7 +744,7 @@ function main() {
725
744
gen_html
726
745
;;
727
746
dockerfile)
728
- gen_dockerfile
747
+ gen_dockerfile ${PYTHON_ABI :- " " }
729
748
;;
730
749
capi)
731
750
cmake_gen ${PYTHON_ABI:- " " }
You can’t perform that action at this time.
0 commit comments