[Fix] get version of CANN on Ascend NPU changed (#234)

open-mmlab · Nov 24, 2023 · b39fbd9 · b39fbd9
1 parent b40fe72
commit b39fbd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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