FROM python:3.8

RUN apt update -y

RUN apt install libgl1 -y # or install opencv opencv-python-headless instead

RUN pip3 install \ 
    torch==1.11.0+cu113 \
    torchvision==0.12.0+cu113 \
    torchaudio==0.11.0+cu113 \
    -f https://download.pytorch.org/whl/cu113/torch_stable.html

COPY requirements.txt /requirements.txt

RUN pip3 install -r requirements.txt

ENTRYPOINT ["bash"]