From 063f7aa7cbacdea393c9b79c036e5d21b5fc3a2a Mon Sep 17 00:00:00 2001 From: Gabriela Cervantes Date: Wed, 12 Jul 2023 16:34:17 +0000 Subject: [PATCH] tests: Add Pytorch Dockerfile This PR adds Pytorch Dockerfile for kata metrics. Signed-off-by: Gabriela Cervantes --- .../pytorch_dockerfile/Dockerfile | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 tests/metrics/machine_learning/pytorch_dockerfile/Dockerfile diff --git a/tests/metrics/machine_learning/pytorch_dockerfile/Dockerfile b/tests/metrics/machine_learning/pytorch_dockerfile/Dockerfile new file mode 100644 index 000000000..7acdd9280 --- /dev/null +++ b/tests/metrics/machine_learning/pytorch_dockerfile/Dockerfile @@ -0,0 +1,19 @@ +# Copyright (c) 2023 Intel Corporation +# +# SPDX-License-Identifier: Apache-2.0 + +# Usage: FROM [image name] +FROM intel/intel-optimized-pytorch:1.12.100 + +# Version of the Dockerfile +LABEL DOCKERFILE_VERSION="1.0" + +RUN apt-get update && \ + apt-get install -y --no-install-recommends build-essential curl git && \ + apt-get remove -y unattended-upgrades && \ + curl -OkL https://github.com/dionhaefner/pyhpc-benchmarks/archive/refs/tags/v3.0.tar.gz && \ + tar -xf v3.0.tar.gz && \ + pip install --no-cache-dir click==8.1.3 && \ + cd pyhpc-benchmarks-3.0 && pip3 install --no-cache-dir --user torch==1.10.0 + +CMD ["/bin/bash"]