diff --git a/cache_replacement/requirements.txt b/cache_replacement/requirements.txt
index 0bd240507b3..539fabd1160 100644
--- a/cache_replacement/requirements.txt
+++ b/cache_replacement/requirements.txt
@@ -48,7 +48,7 @@ tensorboard==2.1.0
 tensorflow==2.1.0
 tensorflow-estimator==2.1.0
 termcolor==1.1.0
-torch==0.4.1.post2
+torch==2.9.1
 tqdm==4.42.1
 urllib3==1.25.8
 Werkzeug==1.0.0