diff --git a/test/images/node-perf/tf-wide-deep/BASEIMAGE b/test/images/node-perf/tf-wide-deep/BASEIMAGE
index 9b23a60fbad..f0cc017f768 100644
--- a/test/images/node-perf/tf-wide-deep/BASEIMAGE
+++ b/test/images/node-perf/tf-wide-deep/BASEIMAGE
@@ -1 +1 @@
-linux/amd64=python:3.6-slim-stretch
+linux/amd64=python:3.11-slim-bookworm
diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile
index a0d69002ef8..6ed7f1fa304 100644
--- a/test/images/node-perf/tf-wide-deep/Dockerfile
+++ b/test/images/node-perf/tf-wide-deep/Dockerfile
@@ -19,22 +19,15 @@ CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/
 
 RUN apt-get update && apt-get install -y wget time
 
-RUN case $(uname -m) in \
-    aarch64) \
-      pip install tensorflow-aarch64==1.2; \
-      ;; \
-    *) \
-      pip install tensorflow==1.9.0; \
-      ;; \
-    esac
+RUN pip install tensorflow==2.20.0
 
-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 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
 
-RUN python /models-1.9.0/official/wide_deep/data_download.py
+RUN python /models-2.15.0/official/wide_deep/data_download.py
 
-WORKDIR $HOME/models-1.9.0/official/wide_deep
-ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0
+WORKDIR $HOME/models-2.15.0/official/wide_deep
+ENV PYTHONPATH=$PYTHONPATH:$HOME/models-2.15.0
 
 ENTRYPOINT python ./wide_deep.py
diff --git a/test/images/node-perf/tf-wide-deep/VERSION b/test/images/node-perf/tf-wide-deep/VERSION
index 7e32cd56983..c068b2447cc 100644
--- a/test/images/node-perf/tf-wide-deep/VERSION
+++ b/test/images/node-perf/tf-wide-deep/VERSION
@@ -1 +1 @@
-1.3
+1.4