From 4d122e2d942829d1ddbe89c79f62111c0bcf67db Mon Sep 17 00:00:00 2001 From: VectorSL Date: Fri, 16 Sep 2022 09:56:08 +0800 Subject: [PATCH] change gpu so path to check version --- mindspore/python/mindspore/run_check/_check_version.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/python/mindspore/run_check/_check_version.py b/mindspore/python/mindspore/run_check/_check_version.py index b1d341b96e0..5fb7626e846 100644 --- a/mindspore/python/mindspore/run_check/_check_version.py +++ b/mindspore/python/mindspore/run_check/_check_version.py @@ -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