diff --git a/metrics/machine_learning/tensorflow_ai_dockerfile/Dockerfile b/metrics/machine_learning/tensorflow_ai_dockerfile/Dockerfile new file mode 100644 index 000000000..a7672bd7f --- /dev/null +++ b/metrics/machine_learning/tensorflow_ai_dockerfile/Dockerfile @@ -0,0 +1,21 @@ +# 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 install -y python3.8 python3-pip && \ + pip install intel-tensorflow-avx512==2.8.0 && \ + pip install protobuf==3.20.* && \ + wget 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"]