diff --git a/init/requirements_pytorch_cpu.txt b/init/requirements_pytorch_cpu.txt
index 215403fb..3c4c9733 100644
--- a/init/requirements_pytorch_cpu.txt
+++ b/init/requirements_pytorch_cpu.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.0+cpu
+torch==2.2.0
 torchvision==0.9.0+cpu
diff --git a/init/requirements_pytorch_cuda.txt b/init/requirements_pytorch_cuda.txt
index a959756b..7f446f59 100644
--- a/init/requirements_pytorch_cuda.txt
+++ b/init/requirements_pytorch_cuda.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.0+cu111
+torch==2.2.0
 torchvision==0.9.0+cu111