Currently this is making use of pynvml in a few places:
|
Alternatively, MIG instance names can be obtained programatically using [NVML](https://developer.nvidia.com/nvidia-management-library-nvml) or [PyNVML](https://github.com/gpuopenanalytics/pynvml). Please refer to the [NVML API](https://docs.nvidia.com/deploy/nvml-api/) to write appropriate utilities for that purpose. |
|
import pynvml |
|
|
|
pynvml.nvmlInit() |
|
return pynvml.nvmlDeviceGetName(pynvml.nvmlDeviceGetHandleByIndex(0)) |
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:deployment/source/guides/mig.md
Line 33 in 3b6a991
deployment/source/cloud/azure/azure-vm-multi.md
Lines 57 to 60 in 3b6a991
However we would like to move to
nvidia-ml-pyin the future. Raising this issue to track this work