Skip to content

Releases: fbcotter/py3nvml

Update py3smi and grab_gpus

11 Oct 13:39
Compare
Choose a tag to compare

Minor update.

  • Fixed some alignment issues with long PIDs in py3smi.
  • Added ability to call py3nvml.grab_gpus with num_gpus=-1. This will grab all available GPUs. Could previously do this by setting num_gpus to a large number but this would throw a warning if it couldn't get all available gpus.

py3nvml.py3nvml updates

04 Mar 18:33
Compare
Choose a tag to compare

To try and keep py3nvml somewhat up-to-date with the constantly evolving nvidia drivers, I have done some work to the py3nvml.py3nvml module. In particular, I have updated all the constants that were missing in py3nvml and existing in the NVIDIA source as of version 418.43. In addition, I have wrapped all of these constants in Enums so it is easier to see what constants go together. Finally, for all the functions in py3nvml.py3nvml I have copied in the C docstring. While this will result in some strange looking docstrings which will be slightly incorrect, they should give good guidance on the scope of the function, something which was ill-defined before.

Finally, I will remove the py3nvml.nvidia_smi module in a future version, as I believe it was only ever meant as an example of how to use the nvml functions to query the gpus, and is now quite out of date. To get the same functionality, you can call nvidia-smi -q -x from python with subprocess.

0.2.2

23 Oct 11:52
Compare
Choose a tag to compare
Added get_free_gpus function and made python2 compatible

Improved py3smi script

27 Jun 17:38
Compare
Choose a tag to compare

Updated script to not use a big xml query but rather multiple small queries. Can now handle if a gpu falls off the bus too, still displaying info for the remaining gpus.

0.2.0

17 May 12:46
Compare
Choose a tag to compare

Added py3smi script

Initial Release

21 Nov 01:22
Compare
Choose a tag to compare
Initial Release Pre-release
Pre-release
Initial Commit

Ran 2to3 on the nvidia-ml-py package from PyPi. Packaged it with
setuptools.