Moving from pynvml to nvidia-ml-py

[jakirkham]

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

ucx-py/ucp/__init__.py

Lines 25 to 28 in 1ef7594

	try:
	import pynvml
	except ImportError:
	pynvml = None

ucx-py/ucp/__init__.py

Lines 51 to 59 in 1ef7594

	def _is_mig_device(handle):
	try:
	pynvml.nvmlDeviceGetMigMode(handle)[0]
	except pynvml.NVMLError:
	return False
	return True
	for dev_idx in range(device_count):
	handle = pynvml.nvmlDeviceGetHandleByIndex(dev_idx)

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

[wence-]

Are the interfacing identical, just the imports differ?

[jakirkham]

Have not looked deeply, but the imports look the same. Maybe there are subtle differences.