FROM ubuntu:20.04
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update
RUN apt-get install -y python3-pip
RUN pip install torch==1.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install numpy 
RUN pip install torchvision==0.11.2+cpu  -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install torchinfo matplotlib
RUN pip install tqdm