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

pytorch 1.7 -> 1.8
BindsNET · Apr 25, 2021 · 9a945a3 · 9a945a3
2 parents d3eca0d + 4f0ebfd
commit 9a945a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
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",