We only need those proto files to build elasticdl.proto, so we can only download those files by wget. Because TensorFlow repo is huge which makes the image huge.
|
ENV TF_PATH /tmp/tensorflow |
|
RUN cd /tmp \ |
|
&& git clone --depth=1 --branch v2.2.0-rc0 \ |
|
https://github.com/tensorflow/tensorflow |