fix bug of env-setting while no LD_LIBRARY_PATH

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
pull/8913/head
zhoufeng 4 years ago
parent 1ee1916828
commit ca3e44dac2

@ -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 "

Loading…
Cancel
Save