Python限制內(nèi)存和CPU使用量的方法（Unix系統(tǒng)適用）
問題
你想對(duì)在Unix系統(tǒng)上面運(yùn)行的程序設(shè)置內(nèi)存或CPU的使用限制。
解決方案
resource 模塊能同時(shí)執(zhí)行這兩個(gè)任務(wù)。例如，要限制CPU時(shí)間，可以像下面這樣做：
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
程序運(yùn)行時(shí)，SIGXCPU 信號(hào)在時(shí)間過期時(shí)被生成，然后執(zhí)行清理并退出。
要限制內(nèi)存使用，設(shè)置可使用的總內(nèi)存值即可，如下：
import resource def limit_memory(maxsize): soft, hard = resource.getrlimit(resource.RLIMIT_AS) resource.setrlimit(resource.RLIMIT_AS, (maxsize, hard))
像這樣設(shè)置了內(nèi)存限制后，程序運(yùn)行到?jīng)]有多余內(nèi)存時(shí)會(huì)拋出 MemoryError 異常。
討論
在本節(jié)例子中，setrlimit() 函數(shù)被用來(lái)設(shè)置特定資源上面的軟限制和硬限制。 軟限制是一個(gè)值，當(dāng)超過這個(gè)值的時(shí)候操作系統(tǒng)通常會(huì)發(fā)送一個(gè)信號(hào)來(lái)限制或通知該進(jìn)程。 硬限制是用來(lái)指定軟限制能設(shè)定的最大值。通常來(lái)講，這個(gè)由系統(tǒng)管理員通過設(shè)置系統(tǒng)級(jí)參數(shù)來(lái)決定。 盡管硬限制可以改小一點(diǎn)，但是最好不要使用用戶進(jìn)程去修改。
setrlimit() 函數(shù)還能被用來(lái)設(shè)置子進(jìn)程數(shù)量、打開文件數(shù)以及類似系統(tǒng)資源的限制。 更多詳情請(qǐng)參考 resource 模塊的文檔。
需要注意的是本節(jié)內(nèi)容只能適用于Unix系統(tǒng)，并且不保證所有系統(tǒng)都能如期工作。 比如我們?cè)跍y(cè)試的時(shí)候，它能在Linux上面正常運(yùn)行，但是在OS X上卻不能。
以上就是Python限制內(nèi)存和CPU的使用量的方法的詳細(xì)內(nèi)容，更多關(guān)于Python限制內(nèi)存和CPU的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
這篇文章主要介紹了Python 處理文件的幾種方式，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-10-10
這篇文章主要為大家詳細(xì)介紹了Python Web框架Tornado運(yùn)行和部署的相關(guān)資料，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2016-04-04
Python multiprocessing 進(jìn)程間通信方式實(shí)現(xiàn)
本文主要介紹了Python multiprocessing 進(jìn)程間通信方式實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02
python正則表達(dá)式及使用正則表達(dá)式的例子
正則表達(dá)是用來(lái)匹配字符串，這篇文章給大家介紹了python正則表達(dá)式及正則表達(dá)式的例子，文章給大家提到了正則表達(dá)式語(yǔ)法規(guī)則，感興趣的朋友一起看看吧2018-01-01
這篇文章主要介紹了Python中的反射和描述器一些知識(shí)的匯總，非常的詳細(xì)，有需要的小伙伴可以參考下2018-09-09
今天你表白了嗎? 這篇文章主要為大家分享了python浪漫表白源碼 ，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-04-04
20個(gè)Python中正則表達(dá)式的應(yīng)用與技巧小結(jié)
正則表達(dá)式是一種用來(lái)匹配字符串的模式,它可以在文本中搜索和匹配特定的字符串模式,本文我們將探討?Python?中?re?模塊的應(yīng)用和一些技巧,希望可以幫助您更好地理解和利用正則表達(dá)式2024-04-04
Python標(biāo)準(zhǔn)庫(kù)之日期、時(shí)間和日歷模塊
這篇文章介紹了Python標(biāo)準(zhǔn)庫(kù)之日期、時(shí)間和日歷模塊，文中通過示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-05-05
BeautifulSoup獲取指定class樣式的div的實(shí)現(xiàn)
這篇文章主要介紹了BeautifulSoup獲取指定class樣式的div的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
本文主要介紹了使用Python進(jìn)行簡(jiǎn)單圖像處理的方法以及Python自動(dòng)生產(chǎn)表情包的實(shí)例，具有很好的參考價(jià)值，下面跟著小編一起來(lái)看下吧2017-03-03