Moving from pynvml to nvidia-ml-py

[jakirkham]

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

dask-cuda/dask_cuda/utils.py

Line 13 in a0fda76

import pynvml

dask-cuda/dask_cuda/tests/test_utils.py

Lines 186 to 187 in a0fda76

	pynvml = pytest.importorskip("pynvml")
	pynvml.nvmlInit()

dask-cuda/dependencies.yaml

Line 103 in a0fda76

- pynvml>=11.0.0

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

[wence-]

I think it is sufficient to just change the dependency, since pynvml is identical in interface.