k8s.io/kubernetes@v1.29.3/test/images/node-perf/tf-wide-deep/Dockerfile (about) 1 # Copyright 2018 The Kubernetes Authors. 2 # 3 # Licensed under the Apache License, Version 2.0 (the "License"); 4 # you may not use this file except in compliance with the License. 5 # You may obtain a copy of the License at 6 # 7 # http://www.apache.org/licenses/LICENSE-2.0 8 # 9 # Unless required by applicable law or agreed to in writing, software 10 # distributed under the License is distributed on an "AS IS" BASIS, 11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 # See the License for the specific language governing permissions and 13 # limitations under the License. 14 15 ARG BASEIMAGE 16 FROM $BASEIMAGE 17 18 CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/ 19 20 RUN apt-get update && apt-get install -y wget time 21 22 RUN case $(uname -m) in \ 23 aarch64) \ 24 pip install tensorflow-aarch64==1.2; \ 25 ;; \ 26 *) \ 27 pip install tensorflow==1.9.0; \ 28 ;; \ 29 esac 30 31 RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \ 32 && tar xzf v1.9.0.tar.gz \ 33 && rm -f v1.9.0.tar.gz 34 35 RUN python /models-1.9.0/official/wide_deep/data_download.py 36 37 WORKDIR $HOME/models-1.9.0/official/wide_deep 38 ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0 39 40 ENTRYPOINT python ./wide_deep.py