tensorflow: Add Mobilnet Tensorflow Dockerfile

This PR adds the Mobilnet Tensorflow Dockerfile for kata containers metrics. Fixes kata-containers#5683 Signed-off-by: Gabriela Cervantes <[email protected]>
GabyCT · Jun 6, 2023 · acaed4c · acaed4c
1 parent 1188a01
commit acaed4c
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/metrics/machine_learning/tensorflow_ai_dockerfile/Dockerfile b/metrics/machine_learning/tensorflow_ai_dockerfile/Dockerfile
@@ -0,0 +1,22 @@
+# Copyright (c) 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+# Usage: FROM [image name]
+FROM ubuntu:20.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Version of the Dockerfile
+LABEL DOCKERFILE_VERSION="1.0"
+
+RUN apt-get update && \
+	apt-get install -y --no-install-recommends wget nano curl build-essential git && \
+	apt-get remove -y unattended-upgrades && \
+	apt-get install -y python3.8  python3-pip && \
+	pip install --no-cache-dir intel-tensorflow-avx512==2.8.0 && \
+	pip install --no-cache-dir protobuf==3.20.* && \
+	wget -q https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_8/mobilenet_v1_1.0_224_frozen.pb && \
+	git clone https://github.com/IntelAI/models.git
+
+CMD ["/bin/bash"]