github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-07-17 20:00:07 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #136349 from dims/fix-tf-wide-deep-image

Browse Source 
test/images: fix tf-wide-deep build by using models v1.9.0
This commit is contained in:
Kubernetes Prow Robot
2026-01-21 01:53:26 +05:30
committed by GitHub
parent 8f76dbf79b d1cc61a78b
commit 76f4413931
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/images/node-perf/tf-wide-deep/Dockerfile
View File

@@ -21,13 +21,16 @@ RUN apt-get update && apt-get install -y wget time
RUN pip install tensorflow==2.20.0
RUN wget https://github.com/tensorflow/models/archive/v2.15.0.tar.gz \
&& tar xzf v2.15.0.tar.gz \
&& rm -f v2.15.0.tar.gz
# Use models v1.9.0 which contains the wide_deep implementation.
# The wide_deep directory was removed in later versions of tensorflow/models.
# The v1.9.0 code uses tf.estimator APIs which are still available in TF 2.x.
RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
&& tar xzf v1.9.0.tar.gz \
&& rm -f v1.9.0.tar.gz
RUN python /models-2.15.0/official/wide_deep/data_download.py
RUN python /models-1.9.0/official/wide_deep/data_download.py
WORKDIR $HOME/models-2.15.0/official/wide_deep
ENV PYTHONPATH=$PYTHONPATH:$HOME/models-2.15.0
WORKDIR $HOME/models-1.9.0/official/wide_deep
ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0
ENTRYPOINT python ./wide_deep.py

2
test/images/node-perf/tf-wide-deep/VERSION
View File

@@ -1 +1 @@
1.4
1.5