Python限制內(nèi)存和CPU使用量的方法(Unix系統(tǒng)適用)
問題
你想對在Unix系統(tǒng)上面運(yùn)行的程序設(shè)置內(nèi)存或CPU的使用限制。
解決方案
resource 模塊能同時(shí)執(zhí)行這兩個(gè)任務(wù)。例如,要限制CPU時(shí)間,可以像下面這樣做:
import signalimport resourceimport osdef 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 信號在時(shí)間過期時(shí)被生成,然后執(zhí)行清理并退出。
要限制內(nèi)存使用,設(shè)置可使用的總內(nèi)存值即可,如下:
import resourcedef 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í)會拋出 MemoryError 異常。
討論
在本節(jié)例子中,setrlimit() 函數(shù)被用來設(shè)置特定資源上面的軟限制和硬限制。 軟限制是一個(gè)值,當(dāng)超過這個(gè)值的時(shí)候操作系統(tǒng)通常會發(fā)送一個(gè)信號來限制或通知該進(jìn)程。 硬限制是用來指定軟限制能設(shè)定的最大值。通常來講,這個(gè)由系統(tǒng)管理員通過設(shè)置系統(tǒng)級參數(shù)來決定。 盡管硬限制可以改小一點(diǎn),但是最好不要使用用戶進(jìn)程去修改。
setrlimit() 函數(shù)還能被用來設(shè)置子進(jìn)程數(shù)量、打開文件數(shù)以及類似系統(tǒng)資源的限制。 更多詳情請參考 resource 模塊的文檔。
需要注意的是本節(jié)內(nèi)容只能適用于Unix系統(tǒng),并且不保證所有系統(tǒng)都能如期工作。 比如我們在測試的時(shí)候,它能在Linux上面正常運(yùn)行,但是在OS X上卻不能。
以上就是Python限制內(nèi)存和CPU的使用量的方法的詳細(xì)內(nèi)容,更多關(guān)于Python限制內(nèi)存和CPU的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!
相關(guān)文章:
1. Python獲取抖音關(guān)注列表封號賬號的實(shí)現(xiàn)代碼2. Warning: require(): open_basedir restriction in effect,目錄配置open_basedir報(bào)錯(cuò)問題分析3. php網(wǎng)絡(luò)安全中命令執(zhí)行漏洞的產(chǎn)生及本質(zhì)探究4. 解決Python 進(jìn)程池Pool中一些坑5. php測試程序運(yùn)行速度和頁面執(zhí)行速度的代碼6. Python如何讀寫CSV文件7. 三個(gè)不常見的 HTML5 實(shí)用新特性簡介8. ajax請求添加自定義header參數(shù)代碼9. python利用os模塊編寫文件復(fù)制功能——copy()函數(shù)用法10. 無線標(biāo)記語言(WML)基礎(chǔ)之WMLScript 基礎(chǔ)第1/2頁
