From 4aa1e3bfdf54ab052b185d61965aa61446a5aeae Mon Sep 17 00:00:00 2001
From: Janosh Riebesell <janosh.riebesell@gmail.com>
Date: Wed, 5 Jun 2024 14:04:34 -0400
Subject: [PATCH] torch>=2

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 87c953dd..e8d8ba8e 100644
--- 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",