diff --git a/dockerize/requirements.txt b/dockerize/requirements.txt
index b98e559..491fdfb 100644
--- a/dockerize/requirements.txt
+++ b/dockerize/requirements.txt
@@ -226,7 +226,7 @@ threadpoolctl==3.6.0
 tifffile==2025.5.10
 tokenizers==0.19.1
 tomli @ file:///opt/conda/conda-bld/tomli_1657175507142/work
-torch==2.4.1+cu121
+torch==2.8.0
 torch-model-archiver==0.12.0
 torch-workflow-archiver==0.2.15
 torchaudio==2.4.1+cu121
diff --git a/requirements.txt b/requirements.txt
index 8f9d8e8..d3a3633 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -223,7 +223,7 @@ threadpoolctl==3.6.0
 tifffile==2025.5.10
 tokenizers==0.19.1
 tomli @ file:///opt/conda/conda-bld/tomli_1657175507142/work
-torch==2.4.1+cu121
+torch==2.8.0
 torch-model-archiver==0.12.0
 torch-workflow-archiver==0.2.15
 torchaudio==2.4.1+cu121