使用subprocess包来在python代码实时查看GPU利用率

最新推荐文章于 2024-04-24 13:37:07 发布

重剑DS

最新推荐文章于 2024-04-24 13:37:07 发布

阅读量288

点赞数 1

分类专栏：深度学习文章标签： python 人工智能深度学习

本文链接：https://blog.csdn.net/qq_45732909/article/details/137370189

版权

深度学习专栏收录该内容

37 篇文章 0 订阅

订阅专栏

最近又被GPU利用率问题导致训练不高效的问题搞到了（恼），所以在py使用代码看看是哪出了问题。

import subprocess

def get_gpu_utilization():
    # 运行nvidia-smi命令
    smi_output = subprocess.check_output(['nvidia-smi', '--query-gpu=utilization.gpu', '--format=csv,noheader,nounits']).decode().strip()

    # 可以返回一个列表，其中包含每个GPU的利用率
    gpu_utilization = [int(x) for x in smi_output.split('\n')]

    return gpu_utilization

# 在你想要检查GPU利用率的代码位置调用这个函数
gpu_util = get_gpu_utilization()
print(f"Current GPU Utilization: {gpu_util}%")