github.com/kata-containers/tests@v0.0.0-20240307153542-772105b56064/metrics/machine_learning/pytorch_dockerfile/Dockerfile (about)

     1  # Copyright (c) 2023 Intel Corporation
     2  #
     3  # SPDX-License-Identifier: Apache-2.0
     4  
     5  # Usage: FROM [image name]
     6  FROM intel/intel-optimized-pytorch:1.12.100
     7  
     8  # Version of the Dockerfile
     9  LABEL DOCKERFILE_VERSION="1.0"
    10  
    11  RUN apt-get update && \
    12  	apt-get install -y --no-install-recommends build-essential curl git && \
    13  	apt-get remove -y unattended-upgrades && \
    14  	curl -OkL https://github.com/dionhaefner/pyhpc-benchmarks/archive/refs/tags/v3.0.tar.gz  && \
    15  	tar -xf v3.0.tar.gz && \
    16  	pip install --no-cache-dir click==8.1.3 && \
    17  	cd pyhpc-benchmarks-3.0 && pip3 install --no-cache-dir --user torch==1.10.0
    18  
    19  CMD ["/bin/bash"]