diff --git a/requirements.txt b/requirements.txt
index 3118b61..945d73e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ matplotlib
 numpy
 scipy
 tensorboardX
-torch==1.6.0
+torch==2.2.0
 tqdm
 pillow
 pyyaml