Added CPU dockerfile

This commit is contained in:
Oscar Blue 2022-03-20 16:38:38 +00:00
parent 36a2242b73
commit b2f8b7606a

22
Dockerfile.cpu Normal file
View file

@ -0,0 +1,22 @@
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"]