diff --git a/distributed/diagnostics/nvml.py b/distributed/diagnostics/nvml.py index 99d22937b3e..6548db6a0c8 100644 --- a/distributed/diagnostics/nvml.py +++ b/distributed/diagnostics/nvml.py @@ -317,6 +317,8 @@ def _get_memory_total(h): def _get_name(h): try: return pynvml.nvmlDeviceGetName(h).decode() + except AttributeError: + return pynvml.nvmlDeviceGetName(h) except pynvml.NVMLError_NotSupported: return None