diff --git a/setup.py b/setup.py
index f0a9baf..ccdc6f0 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@
     'tensorboard==2.16.2',
     'termcolor==2.3.0',
     'tensorflow==2.16.1',
-    'torch==2.0.0',
+    'torch==2.7.0',
     'torchvision==0.15.1',
     'tqdm==4.66.3',
     'vit-keras==0.1.2',