限制你的Python程序所能使用的最大内存

如果程序开发不当，限制序可能会出现占用过多内存的使内存情况。特别是限制序在Docker里面，如果Python程序占用太多内存，使内存可能会导致Docker容器死掉。限制序

为了限制Python程序所能使用的使内存最大内存，我们可以使用Python自带的限制序resource模块。

首先获取系统默认的使内存内存soft上限和hard上限：

import resource soft, hard = resource.getrlimit(resource.RLIMIT_AS) 

其中的resource.RLIMIT_AS实际上就是数字5，表示内存资源。限制序而soft限制是使内存一个可以调节的内存使用上限，云服务器提供商hard是限制序一个一旦设定就很难动态调整的内存使用量上限。

在我的使内存电脑上运行效果如下图所示：

一开始soft和hard的值是一样的。并且它们的限制序单位不是比特。而是使内存比比特还小的单位。我的限制序电脑是8GB内存，而8 * 1024 ^ 6 正好约等于这里的香港云服务器hard和soft的值。

接下来，设置当前运行的这个Python程序能够使用的最大内存：

resource.setrlimit(resource.RLIMIT_AS, (最大上限, hard)) 

其中最大上限是一个整数。假设我想限制当前程序最多可以使用500MB的内存，那么这里的最大上限可以设置为：

0.5 * 1024 ^ 6 = 576460752303423488 

所以设置为：

resource.setrlimit(resource.RLIMIT_AS, (576460752303423488, hard)) 

那么当前Python程序使用的内存超过500MB以后，程序就会抛出MemoryError。

最后，这些代码只能在Linux系统上正常工作。Windows和Mac上可能会有问题。 

很赞哦!（9152）