diff --git a/setup.py b/setup.py
index 45f8280..35a5612 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@
       author_email='yxge@link.cuhk.edu.hk',
       url='https://github.com/yxgeee/MMT.git',
       install_requires=[
-          'numpy', 'torch==1.1.0', 'torchvision==0.2.2', 
+          'numpy', 'torch==2.2.0', 'torchvision==0.2.2', 
           'six', 'h5py', 'Pillow', 'scipy',
           'scikit-learn', 'metric-learn'],
       packages=find_packages(),