Update torch version in Dockerfile

TimoLoomets · TimoLoomets · commit d8b1cf829dd6 · 2020-10-30T18:09:01.000Z
diff --git a/Dockerfile b/Dockerfile
@@ -41,8 +41,7 @@ RUN python3.7 -m pip install sympy
 RUN python3.7 -m pip install matplotlib
 
 # pytorch
-RUN python3.7 -m pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN python3.7 -m pip install torchvision>=0.5.0
+RUN python3.7 -m pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
 
 # tensorflow
 RUN python3.7 -m pip install tensorflow
