diff --git a/setup.py b/setup.py
index 7dfce17..12e879b 100644
--- a/setup.py
+++ b/setup.py
@@ -112,7 +112,7 @@ def get_description():
     # https://packaging.python.org/en/latest/requirements.html
     install_requires = [
         'typing;python_version<"3.5"',
-        'torch==1.7.1',
+        'torch==2.2.0',
         'sentencepiece==0.1.95',
         'tensorboard==2.4.1',
         'progressbar2'