From b2f8b7606a47a342a648be0deb030a1126054cd6 Mon Sep 17 00:00:00 2001
From: Oscar Pocock <oscar@oscarpo.co>
Date: Sun, 20 Mar 2022 16:38:38 +0000
Subject: [PATCH] Added CPU dockerfile

---
 Dockerfile.cpu | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 Dockerfile.cpu

diff --git a/Dockerfile.cpu b/Dockerfile.cpu
new file mode 100644
index 0000000..b437e1e
--- /dev/null
+++ 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"]
\ No newline at end of file