Use UUID to get NVML device
When using PMT in a system with more than one NVIDIA GPU, there may be a difference between CUDA device 0
and NVML device 0
(or any other valid device number). To fix this, the UUID of the CUDA device is used to initialize NVML.