diff --git a/requirements.txt b/requirements.txt
index 88ccf2d..91cb2b4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 scikit-learn
 matplotlib
-torch===1.4.0
+torch===2.2.0
 torchvision===0.5.0
 torchsummary
 Pillow