|
|
|
@@ -92,8 +92,8 @@ class GPUEnvChecker(EnvChecker): |
|
|
|
v = version.parse(v) |
|
|
|
v_str = str(v.major) + "." + str(v.minor) |
|
|
|
if v_str not in self.version: |
|
|
|
raise EnvironmentError(f"MindSpore version {__version__} and cuda version {v_str} does not match, " |
|
|
|
"reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
logger.warning(f"MindSpore version {__version__} and cuda version {v_str} does not match, " |
|
|
|
"reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
|
|
|
|
def _check_env(self): |
|
|
|
"""gpu cuda path check""" |
|
|
|
@@ -176,9 +176,8 @@ class AscendEnvChecker(EnvChecker): |
|
|
|
|
|
|
|
v = self._read_version(self.fwk_version) |
|
|
|
if v not in self.version: |
|
|
|
raise EnvironmentError( |
|
|
|
f"MindSpore version {__version__} and Ascend 910 AI software package version {v} does not " |
|
|
|
"match, reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
logger.warning(f"MindSpore version {__version__} and Ascend 910 AI software package version {v} does not " |
|
|
|
"match, reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
|
|
|
|
def set_env(self): |
|
|
|
if not self.tbe_path: |
|
|
|
|