概述

Memory_profiler是一个Python模块,可以监视一个进程的内存消耗,甚至可以一行一行的分析Python程序的内存消耗。

实例1

用@profile修饰你需要监视的函数(如果函数在类里用:@profile(precision=4))

@profile
def func():
    a = [1] * (2 ** 8)
    b = [2] * (2 ** 10)
    del b
    return a

if __name__ == '__main__':
    func()

运行脚本时需传入-m memory_profiler参数

python -m memory_profiler example.py

实例2

导入模块会实时输出

from memory_profiler import profile
import time
@profile
def func():
    a = [1] * (2 ** 8)
    b = [2] * (2 ** 10)
    del b
    return a

if __name__ == '__main__':
    while True:
        func()
        time.sleep(1)

运行脚本时需传入-m memory_profiler参数

python example.py

memory_profiler工具下载地址

安装:python setup.py install

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐