diff --git a/py/requirements.txt b/py/requirements.txt
index 029ddef..907b046 100644
--- a/py/requirements.txt
+++ b/py/requirements.txt
@@ -29,7 +29,7 @@ thop == 0.0.31.post2004272040
 # lib/utils/util.py: 13
 # resnet/train_resnet.py: 13,14,15,16
 # resnet/train_resnet_v2.py: 13,14,15,16
-torch == 1.4.0
+torch == 2.2.0
 
 # lib/data/pascal_voc.py: 14
 # lib/models/densenet/densenet_121.py: 12