diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index 1f4359f6..e3f91282 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -1,4 +1,4 @@
-torch >= 1.9, < 1.13
+torch == 2.0
 numpy
 tqdm
 requests