Использование мощности графического процессора Nvidia для определенного процесса в Python в Linux

#python #gpu #nvidia

Вопрос:

Есть ли какой-либо способ проверить коэффициент использования мощности определенного процесса графического процессора Nvidia в Python? (Linux)

Nvidia-smi обеспечивает использование графического процессора в целом, и поскольку я выполняю запрос мощности в потоке для процесса, я хочу знать, сколько этот процесс использует всей мощности графического процессора, чтобы я мог извлечь соответствующие данные с помощью nvmlDeviceGetPowerUsage. Для процессора (AMD) Я использую psutil с PID процесса, но есть ли способ/ библиотека, чтобы найти это для графического процессора Nvidia в Python?

Он работает на сервере, и мы-несколько человек, которые его используют, поэтому я не хочу, чтобы энергопотребление в целом.

Большое спасибо!

Комментарии:

1. Я не знаю ответа, но, вероятно, было бы полезно для других, если бы вы сказали, на какую ОС вы нацелены.