diff --git a/requirements.txt b/requirements.txt
index 397d1d3..764677b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 torchvision==0.2.1
 numpy==1.22.0
-torch==0.4.1
+torch==2.2.0
 pandas==0.23.4
 scikit_learn==0.20.3