diff --git a/DIRE/prediction_plugin/requirements_cpu.txt b/DIRE/prediction_plugin/requirements_cpu.txt
index 9d45d2f..16b42e5 100644
--- a/DIRE/prediction_plugin/requirements_cpu.txt
+++ b/DIRE/prediction_plugin/requirements_cpu.txt
@@ -10,4 +10,4 @@ tqdm
 ujson
 pygments
 # if you want to use the cpu, run pip install --find-links https://download.pytorch.org/whl/torch_stable.html -r requirements.txt
-torch==1.6.0+cpu
+torch==2.2.0