From 24baededc041dc306155ce29344ef3bc68c988f0 Mon Sep 17 00:00:00 2001 From: Gabriela Cervantes Date: Tue, 15 Aug 2023 17:36:48 +0000 Subject: [PATCH] metrics: Add Dockerfile for ResNet50 int8 This PR adds the dockerfile for ResNet50 int8 benchmark. Signed-off-by: Gabriela Cervantes --- .../resnet50_int8_dockerfile/Dockerfile | 21 +++++++++++++++++++ .../tensorflow_resnet50_int8.sh | 4 ++-- 2 files changed, 23 insertions(+), 2 deletions(-) create mode 100644 tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile diff --git a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile new file mode 100644 index 0000000000..9782d72e4b --- /dev/null +++ b/tests/metrics/machine_learning/resnet50_int8_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/resnet50_int8_pretrained_model.pb && \ + git clone https://github.com/IntelAI/models.git + +CMD ["/bin/bash"] diff --git a/tests/metrics/machine_learning/tensorflow_resnet50_int8.sh b/tests/metrics/machine_learning/tensorflow_resnet50_int8.sh index 177a62c980..19c3e1192e 100755 --- a/tests/metrics/machine_learning/tensorflow_resnet50_int8.sh +++ b/tests/metrics/machine_learning/tensorflow_resnet50_int8.sh @@ -11,7 +11,7 @@ SCRIPT_PATH=$(dirname "$(readlink -f "$0")") source "${SCRIPT_PATH}/../lib/common.bash" IMAGE="docker.io/library/resnet50int8:latest" -DOCKERFILE="${SCRIPT_PATH}/resnet50_int8/Dockerfile" +DOCKERFILE="${SCRIPT_PATH}/resnet50_int8_dockerfile/Dockerfile" tensorflow_file=$(mktemp tensorflowresults.XXXXXXXXXX) NUM_CONTAINERS="$1" TIMEOUT="$2" @@ -22,7 +22,7 @@ TESTDIR="${TESTDIR:-/testdir}" dst_dir="/host" src_dir=$(mktemp --tmpdir -d tensorflowresnet50int8.XXXXXXXXXX) MOUNT_OPTIONS="type=bind,src=$src_dir,dst=$dst_dir,options=rbind:ro" -start_script="resnet50_start.sh" +start_script="resnet50int8_start.sh" # CMD points to the script that starts the workload # export DNNL_MAX_CPU_ISA=AVX512_CORE_AMX CMD="export KMP_AFFINITY=granularity=fine,verbose,compact && export OMP_NUM_THREADS=16 && $dst_dir/$start_script"