diff --git a/setup.py b/setup.py
index d1d5bb9..8d16edd 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
     packages=find_packages(),
     install_requires=[
         "numpy>=1.17.3",
-        "torch==1.4.0",
+        "torch==2.2.0",
         "tensorboardX",
         "dumpster @ git+https://github.com/ritchie46/dumpster.git#egg=dumpster",
         "patool~=1.12",