Update pytorch/requirements

ShikharJ · ShikharJ · commit 9020ed311591 · 2020-12-18T09:59:29.000+05:30
diff --git a/pytorch/requirements-cpu.txt b/pytorch/requirements-cpu.txt
@@ -4,6 +4,6 @@ numpy==1.16.4
 pandas==0.23.4
 scikit-learn==0.21.2
 scipy==1.3.0
-torch
-torchvision
+torch==1.4.0
+torchvision==0.5.0
 requests
diff --git a/pytorch/requirements-gpu.txt b/pytorch/requirements-gpu.txt
@@ -4,6 +4,6 @@ numpy==1.16.4
 pandas==0.23.4
 scikit-learn==0.21.2
 scipy==1.3.0
-torch
-torchvision
+torch==1.4.0
+torchvision==0.5.0
 requests
