Merge pull request #497 from BindsNET/hananel

freeze torch to 1.8.1
BindsNET · Jun 24, 2021 · cf4af8d · cf4af8d
2 parents eac5af1 + ac7fa04
commit cf4af8d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,8 +2,9 @@ foolbox
 scipy>=1.1.0
 numpy>=1.14.2
 cython>=0.28.5
-torch>=1.5.1
-torchvision>=0.6.1
+torch==1.8.1
+torchvision==0.9.1
+tensorboardX==2.2
 tqdm>=4.19.9
 setuptools>=39.0.1
 matplotlib>=2.1.0
@@ -14,6 +15,5 @@ scikit_learn>=0.19.1
 opencv-python>=3.4.0.12
 pytest>=3.4.0
 pandas>=0.23.4
-tensorboardX>=1.7
 pre-commit
 
diff --git a/setup.py b/setup.py
@@ -17,9 +17,9 @@
     zip_safe=False,
     install_requires=[
         "numpy>=1.14.2",
-        "torch>=1.5.1",
-        "torchvision>=0.6.1",
-        "tensorboardX>=1.7",
+        "torch==1.8.1",
+        "torchvision==0.9.1",
+        "tensorboardX==2.2",
         "tqdm>=4.19.9",
         "matplotlib>=2.1.0",
         "gym>=0.10.4",