python - 函数中使用的内存量？

Question

我可以使用哪个 python 模块来计算执行函数所花费的内存量？我正在使用memory_profiler，但它显示了算法每一行花费的内存量，在这种情况下，我想要一个显示总花费的内存量。

标签： pythonalgorithmmemory-management

Answer 1

你可以tracemalloc用来做memory_profiler自动做的事情。这有点不友好，但我认为它可以很好地完成您想做的事情。
只需按照下面的代码片段。

import tracemalloc

def myFucn():
    tracemalloc.start()
    ## Your code
    print( tracemalloc..get_traced_memory())
    tracemalloc.stop()

输出以 (current,peak) 的形式给出，即当前内存是代码当前使用的内存，峰值内存是程序执行时使用的最大空间。

代码来自geeksforgeeks。查看更多信息。此外，还有其他几种方法可以跟踪链接中解释的内存。一定要检查一下。