Added CPU dockerfile

2022-03-20 16:38:38 +00:00 · 2022-03-20 16:38:38 +00:00 · b2f8b7606a
commit b2f8b7606a
parent 36a2242b73
1 changed files with 22 additions and 0 deletions
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@ -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"]