diff --git a/setup.py b/setup.py
index 95305c0..4e3575f 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
     "scipy==1.10.0",
     "Pillow==9.3.0",
     "imagecodecs==2021.8.26",
-    "torch==1.13.1",
+    "torch==2.2.0",
     "torchvision==0.14.1"
   ],
   classifiers=[
diff --git a/trainer/requirements.txt b/trainer/requirements.txt
index 6b0c6d9..4156eda 100644
--- a/trainer/requirements.txt
+++ b/trainer/requirements.txt
@@ -3,6 +3,6 @@ numpy==1.26.4
 imagecodecs==2024.1.1
 scipy==1.13.0
 Pillow==10.3.0
-torch==2.3.0
+torch==2.2.0
 torchvision==0.18.0
 pytest==8.2.0