diff --git a/pyproject.toml b/pyproject.toml
index 9e14e5113d..9210a2e25a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,6 +6,6 @@ requires = [
     "decorator",
     "pyyaml",
     "scipy",
-    "torch-npu == 2.5.1rc1"
+    "torch-npu >= 2.5.1rc1"
 ]
 build-backend = "setuptools.build_meta"
diff --git a/requirements.txt b/requirements.txt
index 9ef2cd0eda..c4cf016625 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,4 @@ pyyaml
 scipy
 setuptools
 setuptools-scm
-torch-npu == 2.5.1rc1
+torch-npu >= 2.5.1rc1