diff --git a/requirements.txt b/requirements.txt
index f7d17c5..ef653ff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,6 +9,6 @@ scikit-learn
 scipy
 tqdm
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.6.0+cpu
+torch==2.2.0
 --find-links https://download.pytorch.org/whl/torch_stable.html
 torchvision==0.7.0+cpu