Dockerfile

C’est un exemple qui illustre un Dockerfile typique permettant d’utiliser Pytorch pour votre tâche.

# 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

Ensuite vous avez besoin de le build et le push dans votre registre de containers :

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