diff --git a/yolo_segmentation_training/requirements.txt b/yolo_segmentation_training/requirements.txt
index c6873a7..2af7ed7 100644
--- a/yolo_segmentation_training/requirements.txt
+++ b/yolo_segmentation_training/requirements.txt
@@ -8,5 +8,5 @@ pafy==0.5.5
 protobuf==4.24.0
 python-dotenv==1.0.1
 roboflow==1.1.24
-torch==2.2.1
+torch==2.8.0
 ultralytics==8.0.196
\ No newline at end of file