diff --git a/requirements.txt b/requirements.txt
index f33832b..a36c88a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,5 +15,5 @@ ipython
 pytest
 tensorboardX
 pyyaml
-torch==1.5.0+cu101
+torch==2.6.0
 torchvision==0.6.0+cu101