diff --git a/smooth-topk/requirements.txt b/smooth-topk/requirements.txt
index 5542be3..b6e1be7 100644
--- a/smooth-topk/requirements.txt
+++ b/smooth-topk/requirements.txt
@@ -1,5 +1,5 @@
 tqdm
-torch==0.3.0
+torch==2.2.0
 future
 scipy
 numpy