Merge pull request #477 from danielgafni/pytorch-version-update

Hananel-Hazan · web-flow · commit 9a945a3d37a2 · 2021-04-25T14:36:43.000Z
pytorch 1.7 -&gt; 1.8
diff --git a/requirements.txt b/requirements.txt
@@ -2,8 +2,8 @@ foolbox
 scipy>=1.1.0
 numpy>=1.14.2
 cython>=0.28.5
-torch==1.7.1
-torchvision==0.8.2
+torch==1.8.1
+torchvision==0.9.1
 tqdm>=4.19.9
 setuptools>=39.0.1
 matplotlib>=2.1.0
diff --git a/setup.py b/setup.py
@@ -17,8 +17,8 @@
     zip_safe=False,
     install_requires=[
         "numpy>=1.14.2",
-        "torch==1.7.1",
-        "torchvision==0.8.2",
+        "torch==1.8.1",
+        "torchvision==0.9.1",
         "tensorboardX>=1.7",
         "tqdm>=4.19.9",
         "matplotlib>=2.1.0",
