diff --git a/notebooks/requirements.txt b/notebooks/requirements.txt
index 68e5ba7..1c24bca 100644
--- a/notebooks/requirements.txt
+++ b/notebooks/requirements.txt
@@ -30,7 +30,7 @@ tensorflow-addons==0.21.0
 tensorflow-datasets==4.9.2; python_version=='3.8'
 tensorflow-datasets==4.9.3; python_version>'3.8'
 tensorflow-hub==0.15.0
-torch==2.2.0
+torch==2.6.0
 torchvision==0.17.0
 transformers==4.38.0
 urllib3==2.2.2
diff --git a/pytorch_requirements.txt b/pytorch_requirements.txt
index c877cb8..9c4c39d 100644
--- a/pytorch_requirements.txt
+++ b/pytorch_requirements.txt
@@ -7,6 +7,6 @@ numpy==1.24.3
 peft==0.5.0
 protobuf==3.20.3
 python-dateutil==2.8.2
-torch==2.2.0
+torch==2.6.0
 torchvision==0.17.0
 transformers[torch]==4.38.0