diff --git a/setup.py b/setup.py
index 27ea6da..c05689c 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
     author_email="opencode@microsoft.com",
     packages=["kge"],
     install_requires=[
-        "torch==1.4.0",
+        "torch==2.2.0",
         "pyyaml",
         "pandas",
         "argparse",