Merge pull request #8792 from GabyCT/topic/updatenhwc

metrics: Use a specific python version to run tensorflow benchmark
kata-containers · Jan 12, 2024 · a7114a3 · a7114a3
2 parents ffcd95b + 12a41f8
commit a7114a3
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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"]