mirrors
/
mim
mirror of https://github.com/open-mmlab/mim.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

main
6V 2023-11-24 10:24:59 +08:00 committed by GitHub
parent b40fe72e2f
commit b39fbd9730
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mim/utils/utils.py
View File

@ -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