File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -405,7 +405,7 @@ function gen_dockerfile() {
405
405
406
406
cat << EOF
407
407
========================================
408
- Generate /paddle /build/Dockerfile ...
408
+ Generate ${PADDLE_ROOT} /build/Dockerfile ...
409
409
========================================
410
410
EOF
411
411
429
429
CMD=' "true"'
430
430
fi
431
431
432
- cat >> /paddle /build/Dockerfile << EOF
432
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
433
433
ADD python/dist/*.whl /
434
434
# run paddle version to install python packages first
435
435
RUN apt-get update &&\
443
443
${DOCKERFILE_CUDNN_DSO}
444
444
${DOCKERFILE_GPU_ENV}
445
445
ENV NCCL_LAUNCH_MODE PARALLEL
446
- ADD go/cmd/pserver/pserver /usr/bin/
447
- ADD go/cmd/master/master /usr/bin/
446
+ EOF
447
+ if [[ ${WITH_GOLANG:- OFF} == " ON" ]]; then
448
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
449
+ ADD go/cmd/pserver/pserver /usr/bin/
450
+ ADD go/cmd/master/master /usr/bin/
451
+ EOF
452
+ fi
453
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
448
454
# default command shows the paddle version and exit
449
455
CMD [${CMD} ]
450
456
EOF
You can’t perform that action at this time.
0 commit comments