diff --git a/requirements.txt b/requirements.txt
index 26b8d5a..333e929 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 matplotlib==2.2.2
-torch==0.4.1
+torch==2.2.0
 numpy==1.14.3
 pandas==0.22.0