diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index 6a92a42036..fb4541e5d9 100644 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -222,14 +222,11 @@ def get_driver_version(): """ Returns the driver version. """ + # Enable console printing for `hl-smi` check output = subprocess.run( - "hl-smi", - shell=True, - text=True, - stdout=subprocess.PIPE, - stderr=subprocess.PIPE, + "hl-smi", shell=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env={"ENABLE_CONSOLE": "true"} ) - if output.returncode == 0: + if output.returncode == 0 and output.stdout: return version.parse(output.stdout.split("\n")[2].replace(" ", "").split(":")[1][:-1].split("-")[0]) return None