如何在 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

使用 getGPUs 并查询 GPU 类的 load 成员

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