fix(core/cli): fix error messages for CUDA 10.x drivers

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-07-24 20:18:50 +08:00
parent 3ec88f917f
commit 587e2042eb

View file

@ -351,8 +351,7 @@ def main(): # pylint: disable=too-many-branches,too-many-statements,too-many-lo
if (
'nvmlDeviceGetComputeRunningProcesses' in message
or 'nvmlDeviceGetGraphicsRunningProcesses' in message
and Device.cuda_version().startswith('10.')
):
) and Device.cuda_version().startswith('10.'):
message = '\n'.join(
(
message,