Make gpustat.nvml compatible with a third-party fork of pynvml (#153)

An un-official distribution of the pynvml package provided by
@gpuopenanalytics/pynvml was causing an incompatibility issue, where
gpustat.nvml fails to import. Although the official and correct pynvml
packages should be installed, but wild users' python environments can be
easily messed up by unawaringly installing the problematic, unofficial
pynvml package.

gpustat can support this third-party pynvml by directly importing the
low-level pynvml.nvml package (which should be exactly the same as
an official pynvml.py). This is a quickfix that relies on the
implementation detail of a third-party package, but nevertheless it
should be a good enough workaround solution for many users in practice.

(cherry picked from commit 7c09a0f,
 backported to 1.1.x)

Author: wookayin

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)