FROM arm64v8/ubuntu:20.04
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update
RUN apt-get install -y python3-pip
RUN apt-get install -y libopenblas-base libopenmpi-dev
COPY torch-1.10.0a0+git71f889c-cp38-cp38-linux_aarch64.whl /root/
RUN pip install numpy
RUN pip install /root/torch-1.10.0a0+git71f889c-cp38-cp38-linux_aarch64.whl
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
