diff --git a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile index edbdf7370a..6e8de9b0c0 100644 --- a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile +++ b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile @@ -12,10 +12,14 @@ 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 --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/resnet50_fp32_pretrained_model.pb && \ - git clone https://github.com/IntelAI/models.git + git clone https://github.com/IntelAI/models.git && \ + 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 --enable-optimizations && make altinstall && make install && \ + apt-get install -y python3-pip && \ + pip install --no-cache-dir intel-tensorflow-avx512==2.8.0 && \ + pip install --no-cache-dir protobuf==3.20.* CMD ["/bin/bash"] diff --git a/tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh b/tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh index 73ab0a2b22..e6eafc3920 100755 --- a/tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh +++ b/tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh @@ -56,7 +56,7 @@ function create_start_script() { cat <>"${script}" #!/bin/bash -python3.8 models/benchmarks/launch_benchmark.py --benchmark-only --framework tensorflow --model-name resnet50 --precision fp32 --mode inference --in-graph /resnet50_fp32_pretrained_model.pb --batch-size 1 --num-intra-threads 16 >> results +python3.10 models/benchmarks/launch_benchmark.py --benchmark-only --framework tensorflow --model-name resnet50 --precision fp32 --mode inference --in-graph /resnet50_fp32_pretrained_model.pb --batch-size 1 --num-intra-threads 16 >> results EOF chmod +x "${script}" }