torch version

gelles-brandeis · Nov 3, 2022 · ccab59e · ccab59e
1 parent 5b0959d
commit ccab59e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -75,7 +75,7 @@
         "scikit-learn",
         "scipy",
         "tensorboard",
-        "torch==1.11.0",
+        "torch>=1.11.0",
         "typer",
     ],
     extras_require={