diff --git a/requirements.txt b/requirements.txt
index 63bb5a6..538f4a4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,5 @@ matplotlib==3.1.2
 pandas==0.25.3
 scikit-learn==0.21.3
 torchvision==0.4.2
-torch==1.3.1
+torch==2.2.0
 tqdm==4.39.0