Support latest PyTorch release

helmholtz-analytics · Nov 23, 2023 · 626b840 · 626b840
1 parent 5125ccc
commit 626b840
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -34,7 +34,7 @@
     install_requires=[
         "mpi4py>=3.0.0",
         "numpy>=1.20.0",
-        "torch>=1.8.0, <2.0.2",
+        "torch>=1.8.0, <2.1.2",
         "scipy>=0.14.0",
         "pillow>=6.0.0",
         "torchvision>=0.8.0",