Skip to content

Commit

Permalink
Merge pull request #8594 from GabyCT/topic/updatedockerfilet
Browse files Browse the repository at this point in the history
metrics: Update TensorFlow ResNet FP32 dockerfile
  • Loading branch information
dborquez committed Dec 7, 2023
2 parents ad6aab9 + 56dddab commit 298be4a
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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 changes: 1 addition & 1 deletion tests/metrics/machine_learning/tensorflow_resnet50_fp32.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
}
Expand Down

0 comments on commit 298be4a

Please sign in to comment.