python - 在 Python 中测试期间的 CPU 使用率？

Question

我想在测试我的 ML 模型期间(在调用 predict 期间)获取 cpu 使用情况。这是我目前正在做的事情。 p 为当前进程:

start = p.cpu_percent(interval=1)
y_hat = clf.predict(X_test)
print(abs(p.cpu_percent(interval=None) - start)) # prints cpu usage (%)

这是正确的方法还是有更好的方法来实现这一点？

Answer 1

假设您想在您的程序中使用内置函数执行此操作，Python 的 resource 模块可能对您有用 here . psutil 是一个很好的 option (如 Federico 所建议)如果您能够安装软件包。

在您的程序之外，有许多方法可以获取任意进程的 CPU 使用率。如果您在 *nux 上并且更喜欢命令行，top 和类似的命令应该可以完成这项工作。在图形界面上，我个人更喜欢 KSysGuard(我在 Kubuntu 上)。 Gnome 系统监视器也能正常工作。在 Windows 上，任务管理器应该足够了。

编辑:psutil 似乎返回全局用法。如果你只想使用你的进程，你最好使用 resource 或 os.times，但如果你想要总 CPU 利用率(包括其他进程)psutil 是一个更强大的解决方案。

对于 resource 中的 CPU 时间:

import resource

resource.getrusage()[0]  # Returns the time in seconds in user mode
# Note that this time accumulates while the program runs,
# so you might want to save its previous value each time you take a measurement