mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-24 01:24:26 +01:00
959ca49447
This PR adds the TensorFlow ResNet50 fp32 Dockerfile for kata metrics. Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
22 lines
678 B
Docker
22 lines
678 B
Docker
# 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 --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
|
|
|
|
CMD ["/bin/bash"]
|