diff --git a/setup.py b/setup.py
index 0e4eb81..82e6a73 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
     version='1.0',
     description='image retrieval',
     install_requires=[
-        'torch==1.0.0',
+        'torch==2.2.0',
         'torchvision==0.2.1',
     ]
 )