From ca3e44dac2088ffefc6a6e6e1fa6b1cd4910424e Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Mon, 23 Nov 2020 18:30:53 +0800 Subject: [PATCH] fix bug of env-setting while no LD_LIBRARY_PATH Signed-off-by: zhoufeng --- mindspore/_check_version.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index 136cbd8afb..d98eb14caa 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -210,7 +210,10 @@ class AscendEnvChecker(EnvChecker): import te except RuntimeError: if Path(self.tbe_path).is_dir(): - os.environ['LD_LIBRARY_PATH'] = self.tbe_path + ":" + os.environ['LD_LIBRARY_PATH'] + if os.getenv('LD_LIBRARY_PATH'): + os.environ['LD_LIBRARY_PATH'] = self.tbe_path + ":" + os.environ['LD_LIBRARY_PATH'] + else: + os.environ['LD_LIBRARY_PATH'] = self.tbe_path else: raise EnvironmentError( f"No such directory: {self.tbe_path}, Please check if Ascend 910 AI software package is "