File tree Expand file tree Collapse file tree 1 file changed +13
-3
lines changed Expand file tree Collapse file tree 1 file changed +13
-3
lines changed Original file line number Diff line number Diff line change 104
104
# make install should also be test when unittest
105
105
make install -j ` nproc`
106
106
pip install /usr/local/opt/paddle/share/wheels/* .whl
107
- paddle version
107
+ if [[ ${WITH_FLUID_ONLY:- OFF} == " OFF" ]] ; then
108
+ paddle version
109
+ fi
108
110
fi
109
111
}
110
112
183
185
NCCL_DEPS=" "
184
186
fi
185
187
188
+ if [[ ${WITH_FLUID_ONLY:- OFF} == " OFF" ]]; then
189
+ PADDLE_VERSION=" paddle version"
190
+ CMD=' "paddle", "version"'
191
+ else
192
+ PADDLE_VERSION=" true"
193
+ CMD=' "true"'
194
+ fi
195
+
186
196
cat >> /paddle/build/Dockerfile << EOF
187
197
ADD python/dist/*.whl /
188
198
# run paddle version to install python packages first
@@ -192,15 +202,15 @@ EOF
192
202
pip install /*.whl; apt-get install -f -y && \
193
203
apt-get clean -y && \
194
204
rm -f /*.whl && \
195
- paddle version && \
205
+ ${PADDLE_VERSION} && \
196
206
ldconfig
197
207
${DOCKERFILE_CUDNN_DSO}
198
208
${DOCKERFILE_GPU_ENV}
199
209
ENV NCCL_LAUNCH_MODE PARALLEL
200
210
ADD go/cmd/pserver/pserver /usr/bin/
201
211
ADD go/cmd/master/master /usr/bin/
202
212
# default command shows the paddle version and exit
203
- CMD ["paddle", "version" ]
213
+ CMD [${CMD} ]
204
214
EOF
205
215
}
206
216
You can’t perform that action at this time.
0 commit comments