Get GPU Power Usage per PID in python on linux


Is there any way that I can check the power usage utilization rate of a certain process of an NVIDIA Titan RTX in Python? (Linux)

Nvidia-smi gives the GPU utilization as a whole and since I’m doing a power query in a thread for a process I wanna know how much that process uses from the whole GPU power so I can extract accordingly with nvmlDeviceGetPowerUsage. For the CPU (AMD) I use psutil with the process PID but is there a way/ library to find that for Nvidia GPU in Python?

It runs on a server and we’re multiple people who use it that’s why I don’t want the power usage as a whole.

Thanks a lot!

1 Like

Yes, this could be very beneficial for multi-container environments.