diff --git a/setup.py b/setup.py
index df1e1da..09ff580 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,6 @@
     url='https://github.com/locuslab/projected_sinkhorn',
     packages=['projected_sinkhorn'],
     install_requires=[
-        'torch==1.0'
+        'torch==2.2.0'
     ]
 )