運行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())