diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index bfa8d9b29d..8719d6800f 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -232,7 +232,14 @@ class AscendEnvChecker(EnvChecker): self.check_deps_version() if Path(self.op_impl_path).is_dir(): + # python path for sub process + if os.getenv('PYTHONPATH'): + os.environ['PYTHONPATH'] = self.op_impl_path + ":" + os.environ['PYTHONPATH'] + else: + os.environ['PYTHONPATH'] = self.op_impl_path + # sys path for this process sys.path.append(self.op_impl_path) + os.environ['TBE_IMPL_PATH'] = self.op_impl_path else: raise EnvironmentError(