search

rapidsai / rapids-examples

33 stars 24 forks source link

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

Open jakirkham opened 1 year ago

jakirkham commented 1 year ago

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