diff --git a/requirements.txt b/requirements.txt
index 695b2cd..374924d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ ray==1.0.1
 scikit-image==0.17.2
 tensorboard==2.4.0
 tensorflow-gpu==2.3.1
-torch==1.7.0
+torch==2.8.0
 torchvision==0.8.1
 tqdm==4.46.1
 PyOpenGL==3.1.5