Misreported used memory with the driver 535.129.03 about gpustat HOT 3 CLOSED

It's fine and recommended to use the latest version of pynvml (as the python modulename) or nvidia-ml-py (as the PyPI package name to install), assuming NVIDIA does not break the backward compatibility. In general, any latest nvidia-ml-py with the same driver version prefix (e.g. 535.129.03 matches 12.535.133) should work.

To emphasize again, people should:

IMO the pynvml package should be removed from PyPI. I will add this to README, thanks again for reporting.

from gpustat.

Thanks for reporting! I think this is pynvml installation problem. Do you happen to have the wrong pynvml as well as nvidia-ml-py (which conflicts on providing package)? Please confirm by:

pip list | grep nvml

ls -al $(python -c 'import pynvml; print(pynvml.__file__)')`

Try the following if this can fix the issue:

pip uninstall nvidia-ml-py3 pynvml
pip install --force-reinstall --ignore-installed 'nvidia-ml-py'

Maybe I should revert #153 because this is so common error users might encounter unawarely...

from gpustat.

Yes, that fixes the issue. pynvml 11.4.1 was installed. Thanks, Jongwook!!

So what's the general recommendation? Have a particular version of pynvml, or to uninstall it?

Feel free to close this issue if you want.

from gpustat.