diff --git a/requirements.txt b/requirements.txt
index cc6d31d..be0dd4f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 # pip install -r requirements.txt
 numpy == 1.22.0
-torch == 0.4.1
+torch == 2.2.0
 torchvision == 0.2.1
 opencv-python
 visdom