問題

你想對在Unix系統(tǒng)上面運行的程序設置內(nèi)存或CPU的使用限制。

解決方案

resource 模塊能同時執(zhí)行這兩個任務。例如，要限制CPU時間，可以像下面這樣做：

import signal

import resource

import os

def time_exceeded(signo, frame):

print("Time's up!")

raise SystemExit(1)

def set_max_runtime(seconds):

# Install the signal handler and set a resource limit

soft, hard = resource.getrlimit(resource.RLIMIT_CPU)

resource.setrlimit(resource.RLIMIT_CPU, (seconds, hard))

signal.signal(signal.SIGXCPU, time_exceeded)

if __name__ == '__main__':

set_max_runtime(15)

while True:

pass

程序運行時，SIGXCPU信號在時間過期時被生成，然后執(zhí)行清理并退出。

要限制內(nèi)存使用，設置可使用的總內(nèi)存值即可，如下：

import resource

def limit_memory(maxsize):

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

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

像這樣設置了內(nèi)存限制后，程序運行到?jīng)]有多余內(nèi)存時會拋出 MemoryError 異常。

討論

在本節(jié)例子中，setrlimit()函數(shù)被用來設置特定資源上面的軟限制和硬限制。 軟限制是一個值，當超過這個值的時候操作系統(tǒng)通常會發(fā)送一個信號來限制或通知該進程。 硬限制是用來指定軟限制能設定的最大值。通常來講，這個由系統(tǒng)管理員通過設置系統(tǒng)級參數(shù)來決定。 盡管硬限制可以改小一點，但是最好不要使用用戶進程去修改。

setrlimit()函數(shù)還能被用來設置子進程數(shù)量、打開文件數(shù)以及類似系統(tǒng)資源的限制。 更多詳情請參考resource模塊的文檔。

需要注意的是本節(jié)內(nèi)容只能適用于Unix系統(tǒng)，并且不保證所有系統(tǒng)都能如期工作。 比如我們在測試的時候，它能在Linux上面正常運行，但是在OS X上卻不能。

以上就是Python限制內(nèi)存和CPU的使用量的方法的詳細內(nèi)容，更多關于Python限制內(nèi)存和CPU的資料請關注python博客其它相關文章！

總結(jié)

以上是生活随笔為你收集整理的限制python内存上限_Python限制内存和CPU使用量的方法（Unix系统适用）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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