diff --git a/tests/metrics/machine_learning/tensorflow_nhwc_dockerfile/Dockerfile b/tests/metrics/machine_learning/tensorflow_nhwc_dockerfile/Dockerfile
index a8c73d5f61..89f76c7855 100644
--- a/tests/metrics/machine_learning/tensorflow_nhwc_dockerfile/Dockerfile
+++ b/tests/metrics/machine_learning/tensorflow_nhwc_dockerfile/Dockerfile
@@ -11,8 +11,14 @@ LABEL DOCKERFILE_VERSION="1.0"
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && \
-	apt-get install -y --no-install-recommends build-essential git && \
+	apt-get install -y --no-install-recommends build-essential git zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev libsqlite3-dev wget libbz2-dev && \
 	apt-get remove -y unattended-upgrades && \
-	git clone https://github.com/tensorflow/benchmarks
+	git clone https://github.com/tensorflow/benchmarks && \
+	wget -q https://www.python.org/ftp/python/3.8.10/Python-3.8.10.tgz && \
+	tar xzf Python-3.8.10.tgz && \
+	cd Python-3.8.10 && \
+	./configure --prefix=/usr --enable-optimizations && make install && \
+	python -m pip install --no-cache-dir absl-py==1.4.0 && \
+	python -m pip install --no-cache-dir tensorflow==2.15.*
 
 CMD ["/bin/bash"]