python记录程序内存消耗
时间: 2024-01-05 07:19:53 浏览: 180
记录进程cpu和内存使用率的程序(python)
4星 · 用户满意度95%
你可以使用Python内置的`memory_profiler`模块来记录程序的内存消耗。具体步骤如下:
1. 安装`memory_profiler`模块:在终端中运行`pip install memory_profiler`命令来安装该模块。
2. 在需要记录内存消耗的函数或方法前加上`@profile`装饰器。
3. 运行程序时,在终端中使用`mprof run <your_script>`命令来记录内存消耗。这会生成一个`.dat`文件。
4. 使用`mprof plot`命令来生成内存消耗图表。
例如,下面是一个简单的示例代码:
```python
from memory_profiler import profile
@profile
def test_func():
a = [1] * (10 ** 6)
b = [2] * (2 * 10 ** 7)
del b
return a
if __name__ == '__main__':
test_func()
```
在终端中运行`mprof run test.py`命令,程序会执行并记录内存消耗,生成一个名为`mprofile_*.dat`的文件。然后运行`mprof plot`命令,就可以生成内存消耗图表。
阅读全文