Skip to content

Commit 61eb3aa

Browse files
committed
Dockerfile: fix production dockefile when WITH_GOLONG=OFF.
1 parent 0446220 commit 61eb3aa

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

paddle/scripts/paddle_build.sh

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ function gen_dockerfile() {
405405

406406
cat <<EOF
407407
========================================
408-
Generate /paddle/build/Dockerfile ...
408+
Generate ${PADDLE_ROOT}/build/Dockerfile ...
409409
========================================
410410
EOF
411411

@@ -429,7 +429,7 @@ EOF
429429
CMD='"true"'
430430
fi
431431

432-
cat >> /paddle/build/Dockerfile <<EOF
432+
cat >> ${PADDLE_ROOT}/build/Dockerfile <<EOF
433433
ADD python/dist/*.whl /
434434
# run paddle version to install python packages first
435435
RUN apt-get update &&\
@@ -443,8 +443,14 @@ EOF
443443
${DOCKERFILE_CUDNN_DSO}
444444
${DOCKERFILE_GPU_ENV}
445445
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
448454
# default command shows the paddle version and exit
449455
CMD [${CMD}]
450456
EOF

0 commit comments

Comments
 (0)