github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-06-23 14:08:31 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #8594 from GabyCT/topic/updatedockerfilet

Browse Source 
metrics: Update TensorFlow ResNet FP32 dockerfile
This commit is contained in:
David Esparza 2023-12-07 11:14:48 -06:00 committed by GitHub
commit 298be4aa1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
View File

@ -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"]

2
tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh
View File

@ -56,7 +56,7 @@ function create_start_script() {
cat <<EOF >>"${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}"
}