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
Currently this is making use of
pynvmlin a few places:rapids-examples/dask-metrics/dask_metrics/monitor.py
Line 14 in bedd00f
rapids-examples/dask-metrics/README.md
Line 189 in bedd00f
However we would like to move to
nvidia-ml-pyin the future. Raising this issue to track this work