mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-18 23:04:20 +01:00
a864d0e349
This PR adds the tensorflow mobilenet dockerfile. Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
22 lines
642 B
Docker
22 lines
642 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 intel-tensorflow-avx512==2.8.0 && \
|
|
pip install protobuf==3.20.* && \
|
|
wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_8/mobilenet_v1_1.0_224_frozen.pb && \
|
|
git clone https://github.com/IntelAI/models.git
|
|
|
|
CMD ["/bin/bash"]
|