如何跟踪运行Python代码时使用的虚拟内存？

Question

我有一个Python包我正在使用虚拟内存的基准测试。

目前，这不是很精确。我提交脚本，例如

python script1.py

并查看用于执行的时间和资源。

是否有一些软件包/方法可以找出我的RAM瓶颈在哪里？我正在考虑像

这样的工具

ram_used python script1.py 

将跟踪使用多少RAM。

这样的事情存在吗？

Answer 1

是的，有 - 它被称为memory_profiler。一旦您完成并将@profile装饰器添加到您想要分析的功能，您就可以使用;

python -m memory_profiler example.py

Answer 2

还有memory_info psutil。 https://pythonhosted.org/psutil/

放在脚本的开头：

import psutil
p = psutil.Process()

并且假设代码运行正常，请将其放在最后一行：

print(p.memory_info())