如何在python中记录GPU利用率

Question

我一直在使用一个名为GPUtil的库来尝试在阵列中记录我GPU的利用率百分比值。

我使用的命令输出的数据太多，无法添加到我的数组中。我只想要核心的百分比利用率。内存利用率与我无关。

这两个数组用于PyQtgraph中的图形

我编写的函数如下：

import GPUtil
import time
time_x = []
gpu_y = []

def gpu_util_timer(self):
    for n in range(10):
        Graph_Util.gpu_y.append(GPUtil.showUtilization())
        Graph_Util.time_x.append(n)
        time.sleep(1)
    print('gpu done')

此后，我接受了以下建议并将其实施为：

def N_gpu_util_timer(self):
    for n in range(10):
        GPUs = GPUtil.getGPUs()
        gpu_load = GPUs[0].load
        Graph.gpu_y.append(gpu_load)
        time.sleep(1)
    print(Graph.gpu_y)
    print('N gpu done')

Answer 1

请参阅文档

https://github.com/anderskm/gputil#usage

使用getGPU并查询GPU类的加载成员

GPUs = GPUtil.getGPUs()
load = GPUs[0].load