update torch to 1.8.1 for test matrix

nyu-mll · Apr 17, 2021 · 06a62f0 · 06a62f0
1 parent 4226f49
commit 06a62f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 -r requirements-no-torch.txt
-torch>=1.5.0
-torchvision==0.6.0
+torch>=1.8.1
+torchvision==0.9.1
diff --git a/setup.py b/setup.py
@@ -73,10 +73,10 @@
         "scipy == 1.4.1",
         "sentencepiece == 0.1.86",
         "tokenizers == 0.8.1.rc2",
-        "torch >= 1.5.0",
+        "torch >= 1.8.1",
         "tqdm == 4.46.0",
         "transformers == 3.1.0",
-        "torchvision == 0.6.0",
+        "torchvision == 0.9.1",
     ],
     extras_require=extras,
     python_requires=">=3.6.0",