rapids-examples icon indicating copy to clipboard operation
rapids-examples copied to clipboard
verified publisher icon rapidsai

Moving from `pynvml` to `nvidia-ml-py`

Open jakirkham opened this issue 3 years ago • 0 comments

Currently this is making use of pynvml in a few places:

https://github.com/rapidsai/rapids-examples/blob/bedd00ff23d63d07b86a53fc8dae1497b279bc84/dask-metrics/dask_metrics/monitor.py#L14

https://github.com/rapidsai/rapids-examples/blob/bedd00ff23d63d07b86a53fc8dae1497b279bc84/dask-metrics/README.md?plain=1#L189

However we would like to move to nvidia-ml-py in the future. Raising this issue to track this work

jakirkham avatar Feb 14 '23 20:02 jakirkham