diff --git a/mim/utils/utils.py b/mim/utils/utils.py
index f8c41e1..19e5700 100644
--- a/mim/utils/utils.py
+++ b/mim/utils/utils.py
@@ -353,7 +353,7 @@ def get_npu_version() -> str:
                 'installing OpenMMLab projects on Ascend NPU.'
                 "Please run 'source set_env.sh' in the CANN installation path."
             ))
-    npu_version = torch_npu.get_cann_version(ascend_home_path).lower()
+    npu_version = torch.version.cann
     return npu_version