diff --git a/gpustat/nvml.py b/gpustat/nvml.py index 0427bac..fcf684c 100644 --- a/gpustat/nvml.py +++ b/gpustat/nvml.py @@ -27,7 +27,8 @@ raise ImportError("pynvml library is outdated.") if not hasattr(pynvml, '_nvmlGetFunctionPointer'): - raise ImportError("pynvml appears to be a non-official package.") + # Unofficial pynvml from @gpuopenanalytics/pynvml, see #153 + import pynvml.nvml as pynvml except (ImportError, SyntaxError, RuntimeError) as e: _pynvml = sys.modules.get('pynvml', None)