diff --git a/requirements-cpu.txt b/requirements-cpu.txt
index 5128980..ba5b9e1 100644
--- a/requirements-cpu.txt
+++ b/requirements-cpu.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==1.3.1+cpu
+torch==2.2.0
 torchvision==0.4.2+cpu
\ No newline at end of file
diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index 6aea1bd..a14595e 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -1,2 +1,2 @@
-torch==1.5.0
+torch==2.2.0
 torchvision==0.6.0
\ No newline at end of file