Skip to content

Update py3smi and grab_gpus

Latest
Compare
Choose a tag to compare
@fbcotter fbcotter released this 11 Oct 13:39
· 14 commits to master since this release

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.