diff --git a/requirements.txt b/requirements.txt
index 458c8bf..dc1884d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ tqdm
 matplotlib
 scikit-learn
 pandas
-torch==1.4.0
+torch==2.2.0
 scipy
 # local package
 # -e .