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.10.2+cpu \
torchvision==0.11.3+cpu \
torchaudio==0.10.2+cpu \
-f https://download.pytorch.org/whl/cpu/torch_stable.html
COPY requirements.txt /requirements.txt
RUN pip3 install -r requirements.txt
ENTRYPOINT ["bash"]
