diff --git a/baselines/code/setup.py b/baselines/code/setup.py
index 0e2e9e4..a20293e 100644
--- a/baselines/code/setup.py
+++ b/baselines/code/setup.py
@@ -41,7 +41,7 @@ def get_nmtpytorch_version():
     keywords='nmt neural-mt translation deep-learning pytorch captioning',
     install_requires=[
         'numpy', 'scipy', 'scikit-learn', 'tqdm', 'pillow',
-        'torch==0.3.1', 'torchvision==0.2.1',
+        'torch==2.2.0', 'torchvision==0.2.1',
         'sacrebleu>=1.2.9', 'tensorboardX==1.1',
         'editdistance==0.4', 'ipdb',
     ],