allow torch version

mir-group · Nov 18, 2021 · 8c80a3d · 8c80a3d
1 parent bd99b3f
commit 8c80a3d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -13,7 +13,7 @@ NequIP is an open-source code for building E(3)-equivariant interatomic potentia
 NequIP requires:
 
 * Python >= 3.6
-* PyTorch >= 1.8, < 1.10
+* PyTorch >= 1.8, <= 1.10.*
 
 To install:
 

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
         "numpy",
         "ase",
         "tqdm",
-        "torch>=1.8,<1.10",  # torch.fx added in 1.8
+        "torch>=1.8,<1.11",  # torch.fx added in 1.8
         "e3nn>=0.3.5,<0.5.0",
         "pyyaml",
         "contextlib2;python_version<'3.7'",  # backport of nullcontext