torch>=2

CederGroupHub · Jun 5, 2024 · 4aa1e3b · 4aa1e3b
1 parent b0c3f12
commit 4aa1e3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "numpy>=1.26",
     "nvidia-ml-py3>=7.352.0",
     "pymatgen>=2024.6.4",
-    "torch>=1.11.0",
+    "torch>=2",
 ]
 classifiers = [
     "Intended Audience :: Science/Research",