Retrieve UUIDs of the GPUs used by a worker

Ray Core
1

To profile GPU utilization and power usage per trial/worker I need to match the GPU each worker has access for. For that I try to use UUIDs as I can get GPU stats in amdsmi and pynvml by the GPU’s UUID.

Is there any possible way to get each worker’s GPU UUID or similar?

2

Hi @lineick, have you looked into ray.get_gpu_ids()? ray.get_gpu_ids — Ray 2.47.1 Maybe this will help?