diff --git a/requirements.txt b/requirements.txt
index 9f3b090..56956cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 numpy==1.18.1
-torch==1.6.0
+torch==2.2.0
 scikit_learn==0.23.1