彭博社開源 Memray,一個 Python 內存剖析器
Memray 是一個由彭博社開發的內存剖析器memory profiler,現在已經開源。它可以跟蹤 Python 代碼中的內存分配,包括本地擴展和 Python 解釋器本身。內存剖析是了解程序如何利用內存的有力工具,因此可以檢測內存泄漏或確定程序中哪些區域消耗的內存最多。
與 py-spy 等抽樣內存剖析器相比,Memray 可以跟蹤每個函數調用,包括對 C/C++ 庫的調用,并詳細顯示調用棧。彭博社稱,這并不以犧牲性能為代價,剖析只使解釋代碼的速度變慢一點。然而,原生代碼剖析的速度較慢,因此需要直接啟用。
Memray 可以根據獲得的內存消耗數據生成各種報告,包括火焰圖,這對快速、準確地識別最常見的代碼路徑很有價值。
據 EgdeDB 的聯合創始人兼 CEO Yury Selivanov 稱,該工具提供了以前無法獲得的對 Python 應用的洞察力。Memray 可以用來從命令行中執行和剖析 Python 應用。
$ python3 -m memray run -o output.bin my_script.py
$ python3 -m memray flamegraph output.bin
另外,你可以使用 pytest-memray 將 Memray 集成到你的測試套件中。你也可以用 -native 命令行選項對所有的 C/C++ 調用進行剖析,或者用 -live 命令行選項在程序執行過程中實時分析內存分配。Memray 可以在 Linux x86/64 系統上用 python3 -m pip install memray 來安裝。