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

本文主要是介绍使用subprocess包来在python代码实时查看GPU利用率，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

import subprocessdef 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}%")

这篇关于使用subprocess包来在python代码实时查看GPU利用率的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---