Adding the stable torch wheel to reduce download requirement when bui…

…lding and adding dev deps
dstl · Aug 25, 2022 · 5d60ce0 · 5d60ce0
1 parent be6b2ba
commit 5d60ce0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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