Merge pull request #6 from br0kej/docker-container

Adding a CPU Capable Docker Container
dstl · Sep 9, 2022 · 6457037 · 6457037
2 parents 41511a7 + 5d60ce0
commit 6457037
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -0,0 +1,3 @@
+venv/
+.pytest_cache
+*.egg-info
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -0,0 +1,7 @@
+FROM python:3.8.13-slim
+
+COPY ./ /root/code
+RUN apt-get update && apt-get install gcc -y
+RUN cd /root/code && python3 -m pip install --upgrade pip && pip install torch==1.11+cpu -f https://download.pytorch.org/whl/torch_stable.html && python3 -m pip install -e .[dev]
+
+CMD /bin/bash