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 @@ -398,7 +398,7 @@ function gen_dockerfile() {
398
398
399
399
cat << EOF
400
400
========================================
401
- Generate /paddle /build/Dockerfile ...
401
+ Generate ${PADDLE_ROOT} /build/Dockerfile ...
402
402
========================================
403
403
EOF
404
404
422
422
CMD=' "true"'
423
423
fi
424
424
425
- cat >> /paddle /build/Dockerfile << EOF
425
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
426
426
ADD python/dist/*.whl /
427
427
# run paddle version to install python packages first
428
428
RUN apt-get update &&\
436
436
${DOCKERFILE_CUDNN_DSO}
437
437
${DOCKERFILE_GPU_ENV}
438
438
ENV NCCL_LAUNCH_MODE PARALLEL
439
- ADD go/cmd/pserver/pserver /usr/bin/
440
- ADD go/cmd/master/master /usr/bin/
439
+ EOF
440
+ if [[ ${WITH_GOLANG:- OFF} == " ON" ]]; then
441
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
442
+ ADD go/cmd/pserver/pserver /usr/bin/
443
+ ADD go/cmd/master/master /usr/bin/
444
+ EOF
445
+ fi
446
+ cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
441
447
# default command shows the paddle version and exit
442
448
CMD [${CMD} ]
443
449
EOF
You can’t perform that action at this time.
0 commit comments