diff --git a/dlrover/trainer/torch/utils.py b/dlrover/trainer/torch/utils.py
index c2ee9ec2b..185245de6 100644
--- a/dlrover/trainer/torch/utils.py
+++ b/dlrover/trainer/torch/utils.py
@@ -16,4 +16,4 @@
 
 
 def version_less_than_230():
-    return Version(torch.__version__) < Version("2.3.0")
+    return Version(torch.__version__) <= Version("2.2.2")