函数中使用的内存量？

Question

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

Answer 1

您可以使用tracemalloc 来执行memory_profiller 自动执行的操作。这有点不友好，但我认为它可以很好地完成您想做的事情。
只需按照下面的代码 sn-p 即可。

import tracemalloc

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

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

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