Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
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)
- Loading branch information