diff --git a/py/requirements.txt b/py/requirements.txt
index 5b0ed7f..34ba93c 100644
--- a/py/requirements.txt
+++ b/py/requirements.txt
@@ -25,7 +25,7 @@ numpy == 1.17.2
 # lib/train.py: 12,13,14
 # lib/utils/file.py: 13
 # lib/utils/util.py: 11
-torch == 1.4.0
+torch == 2.2.0
 
 # batch_detect.py: 15
 # detector.py: 14