diff --git a/requirements.txt b/requirements.txt
index 4740669..d276899 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 numpy==1.14.0
-torch==0.4.1
+torch==2.2.0
 git+git://github.com/huggingface/pytorch-pretrained-BERT@3fc63f126ddf883ba9659f13ec046c3639db7b7e