numba.cuda.select_device()
allows device selection by numeric ID, which can be arbitrarily reset via CUDA_VISIBLE_DEVICES
, is there a way to select the device by UUID instead? That can be shown by nvidia-smi -L
and is unique.
If not, is there at least a way of getting the UUID of the GPU that numba
is currently running on? There is cuda.gpus.current.name
, but again, that seems to return the ID.