mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!42116 master apply the path change of mindspore_gpu so for checkversion 

Merge pull request !42116 from VectorSL/change-so-path
This commit is contained in:
i-robot 2022-09-16 06:27:11 +00:00 committed by Gitee
parent 3ebaecdbb1 4d122e2d94
commit 51da8fbbbb
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/python/mindspore/run_check/_check_version.py
View File

@ -76,7 +76,7 @@ class GPUEnvChecker(EnvChecker):
"installation guidelines: https://www.mindspore.cn/install")
else:
logger.warning(f"MindSpore version {__version__} and cuda version {self.v} does not match, "
"CUDA version [{self.version}] are supported by MindSpore officially. "
f"CUDA version [{self.version}] are supported by MindSpore officially. "
"Please refer to the installation guide for version matching "
"information: https://www.mindspore.cn/install.")
nvcc_version = self._get_nvcc_version(False)
@ -175,12 +175,12 @@ class GPUEnvChecker(EnvChecker):
"""Get gpu lib path by ldd command."""
path_list = []
current_path = os.path.split(os.path.realpath(__file__))[0]
mindspore_path = os.path.join(current_path, "../")
mindspore_path = os.path.join(current_path, "../lib/plugin")
try:
real_path = glob.glob(mindspore_path + "/_c_expression*.so*")
real_path = glob.glob(mindspore_path + "/libmindspore_gpu.so")
if real_path == []:
logger.error(f"{self.lib_key_to_lib_name[lib_name]} (need by mindspore-gpu) is not found, please "
f"confirm that _c_expression.so is in directory:{mindspore_path} and the correct cuda "
f"confirm that libmindspore_gpu.so is in directory:{mindspore_path} and the correct cuda "
"version has been installed, you can refer to the installation "
"guidelines: https://www.mindspore.cn/install")
return path_list