From b39fbd97303574a77f6acaed923fad41c781691c Mon Sep 17 00:00:00 2001
From: 6V <49866079+6Vvv@users.noreply.github.com>
Date: Fri, 24 Nov 2023 10:24:59 +0800
Subject: [PATCH] [Fix] get version of CANN on Ascend NPU changed (#234)

---
 mim/utils/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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