diff --git a/custom-nb-image/requirements/requirements.txt b/custom-nb-image/requirements/requirements.txt
index 35b5d559..fb5589c0 100644
--- a/custom-nb-image/requirements/requirements.txt
+++ b/custom-nb-image/requirements/requirements.txt
@@ -183,7 +183,7 @@ threadpoolctl==3.1.0; python_version >= '3.6'
 tinycss2==1.1.1; python_version >= '3.6'
 tomli==2.0.1; python_full_version < '3.11.0a7'
 toolz==0.12.0
-torch==1.12.1
+torch==2.2.0
 torchvision==0.13.1
 tornado==6.2; python_version >= '3.7'
 tqdm==4.64.1; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'