Dockerfile
==========

Here is an example of a typical :code:`Dockerfile` for using `Pytorch <https://pytorch.org/>`_ in your task.

.. code:: dockerfile

    # 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:

.. code:: bash

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