python - 达到内存限制时退出 Python 程序

Question

我有几个 Python/Numpy 程序，当它们使用太多内存时，它们往往会导致 PC 卡住/运行非常缓慢。当它使用大量内存(例如 3.8/4GB)时，我什至无法停止脚本或移动光标因此，我想在程序达到内存使用的临界限制时自动退出该程序，例如3GB。

我还没有找到解决办法。有没有一种 Pythonic 的方法来处理这个问题，因为我在 Windows 和 Linux 机器上运行我的脚本。

Answer 1

您可以限制进程的内存限制，但这是特定于操作系统的。

另一种解决方案是检查 psutil.virtual_memory() 的值，并在达到某个点时退出您的程序。

虽然独立于操作系统，但第二种解决方案根本不是 Pythonic。内存管理是我们拥有操作系统的目的之一。