Dockerfile

Here is an example of a typical Dockerfile for using Pytorch in your task.

# Use an official Python runtime as a parent image
FROM hugomir/manta_light:latest AS build

RUN pip3 install --upgrade pip
RUN pip3 install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu

Then, you can build and push the image to your container registry:

docker build -t <your_tag> .
docker image push <your_tag>