如何调试Python 程序的内存泄露问题

摄影：产品经理

产品经理中的霸王餐

如果大家在 Linux 或者 macOS 下面运行一段可能导致内存泄露的程序，那么你可能会看到下面这样的情况：

而如果你用的系统是 Windows，那么可能电脑直接就卡死了。

但是，调试这种 OOM（Out of Memory）的问题有时候是非常困难的，因为你不知道代码哪个地方会导致内存泄露。但是如果你运行程序进行调试，程序又会中途被杀掉或者直接卡死系统。

如果我们有办法看到程序里面每一个函数占用的内存大小，那么我们就能缩小检查的范围。

为了实现这个目的，我们可以安装并使用一个叫做filprofiler的第三方库，它可以分析 Python 程序的内存占用情况。

我们先安装这个库：

pip install filprofiler

然后写一段会导致内存泄露的代码：

def func_a():
    print('我是一个正常的函数')

def func_b():
    print('我是第二个正常的函数')

def func_b():
    print('我是第三个正常的函数')

def func_oom():
    print('我是一个会导致内存泄露的函数')
    datas = []
    while True:
        datas.append('s' * 1024 * 1024)
    print('运行程序的时候，你不会看到这一行')


def run():
    func_a()
    func_b()
    func_oom()


run()

这段程序直接运行会因为内存泄露的问题被系统直接杀死。

在使用filprofiler之前，还需要调整一下虚拟内存的大小。否则，filprofiler本身也会因为占用内存过大的问题而被系统杀掉。

先使用free命令看一下系统可用的内存有多少：

系统可用内存为1619456 KB

我们使用ulimit命令，把程序能够使用的内存稍稍调低一些，这样即使被占满，也不会被系统杀死：

ulimit -Sv 1600000

然后，使用 filprofiler 来运行这个程序：

fil-profile run test.py

运行效果如下图所示：

filprofiler 会在当前文件夹下面生成一个fil-result文件夹，在里面会有一个以时间命名的文件夹，文件夹中会有两个svg文件，如下图所示：

我们使用浏览器打开其中的out-of-memory.svg文件，可以看到如下图所示的内存占用图：

从图中可以看到，占用内存最大的函数是func_oom，程序也是在这个地方崩溃的。