From 5d60ce01c2f321bf4a64fe745a2ec1b0725a2d98 Mon Sep 17 00:00:00 2001
From: Br0kej <br0kej@protonmail.com>
Date: Thu, 25 Aug 2022 21:53:22 +0100
Subject: [PATCH] Adding the stable torch wheel to reduce download requirement
 when building and adding dev deps

---
 Dockerfile.cpu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 8f7b5125..edf927b1 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -2,6 +2,6 @@ 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 && python3 -m pip install -e .
+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
\ No newline at end of file