FROM python:3.8

RUN apt update -y

RUN apt install libgl1 -y

RUN pip3 install \ 
    torch==1.10.2+cpu \ 
    torchvision==0.11.3+cpu \
    torchaudio==0.10.2+cpu \
    -f https://download.pytorch.org/whl/cpu/torch_stable.html

RUN pip3 install \
    opencv-python \
    sklearn \
    matplotlib \
    pandas \
    tqdm 

#COPY src /src

ENTRYPOINT ["bash"]