diff --git a/requirements.txt b/requirements.txt
index 3190a96..997945d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,4 @@ numpy == 1.18.1
 scikit-learn == 0.22.1
 
 # just for test
-torch == 1.4.0
+torch == 2.2.0