diff --git a/requirements.txt b/requirements.txt
index e1fe81f..d9315d1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.6.0
+torch==2.7.1
 transformers>=4.47.0
 accelerate==0.26.0
 fschat==0.2.31
diff --git a/setup.py b/setup.py
index f83163d..6322910 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
     url='https://github.com/SafeAILab/EAGLE',
     packages=find_packages(),
     install_requires=[
-        'torch==2.0.1',
+        'torch==2.7.1',
         'transformers==4.46.2',
         'accelerate==0.21.0',
         'fschat==0.2.31',