17 lines
386 B
Text
17 lines
386 B
Text
|
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"]
|