https://mp.weixin.qq.com/s/Bi7f6VwT8DoXbuozlX3PIw
限制程序最大內(nèi)存使用

如果程序開發(fā)不當(dāng)，可能會(huì)出現(xiàn)占用過(guò)多內(nèi)存的情況。特別是在Docker里面，如果Python程序占用太多內(nèi)存，可能會(huì)導(dǎo)致Docker容器死掉。

為了限制Python程序所能使用的最大內(nèi)存，我們可以使用Python自帶的resource模塊。

首先獲取系統(tǒng)默認(rèn)的內(nèi)存soft上限和hard上限：

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

其中的resource.RLIMIT_AS實(shí)際上就是數(shù)字5，表示內(nèi)存資源。而soft限制是一個(gè)可以調(diào)節(jié)的內(nèi)存使用上限，hard是一個(gè)一旦設(shè)定就很難動(dòng)態(tài)調(diào)整的內(nèi)存使用量上限。

在我的電腦上運(yùn)行效果如下圖所示：

一開始soft和hard的值是一樣的。并且它們的單位不是比特。而是比比特還小的單位。我的電腦是8GB內(nèi)存，而8 * 1024 ^ 6 正好約等于這里的hard和soft的值。

接下來(lái)，設(shè)置當(dāng)前運(yùn)行的這個(gè)Python程序能夠使用的最大內(nèi)存：
resource.setrlimit(resource.RLIMIT_AS, (最大上限, hard))
其中最大上限是一個(gè)整數(shù)。假設(shè)我想限制當(dāng)前程序最多可以使用500MB的內(nèi)存，那么這里的最大上限可以設(shè)置為：
0.5 * 1024 ^ 6 = 576460752303423488
所以設(shè)置為：
resource.setrlimit(resource.RLIMIT_AS, (576460752303423488, hard))
那么當(dāng)前Python程序使用的內(nèi)存超過(guò)500MB以后，程序就會(huì)拋出MemoryError。

最后，這些代碼只能在Linux系統(tǒng)上正常工作。Windows和Mac上可能會(huì)有問(wèn)題。

總結(jié)

以上是生活随笔為你收集整理的python系统问题的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。