Python限制內(nèi)存和CPU使用量的方法（Unix系統(tǒng)適用）

更新時(shí)間：2020年08月04日 16:27:33 作者：David Beazley

這篇文章主要介紹了Python限制內(nèi)存和CPU的使用量的方法，文中講解非常細(xì)致，代碼幫助大家更好的理解和學(xué)習(xí)，感興趣的朋友可以了解下

問題

你想對在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 信號在時(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í)會拋出 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)注腳本之家其它相關(guān)文章！

您可能感興趣的文章:

相關(guān)文章

Pytorch寫數(shù)字識別LeNet模型
這篇文章主要介紹了Pytorch寫數(shù)字識別LeNet模型，LeNet-5是一個(gè)較簡單的卷積神經(jīng)網(wǎng)絡(luò)，??LeNet-5?這個(gè)網(wǎng)絡(luò)雖然很小，但是它包含了深度學(xué)習(xí)的基本模塊：卷積層，池化層，全連接層。是其他深度學(xué)習(xí)模型的基礎(chǔ)，?這里我們對LeNet-5進(jìn)行深入分析,需要的朋友可以參考下
2022-01-01
Pytorch實(shí)現(xiàn)網(wǎng)絡(luò)部分層的固定不進(jìn)行回傳更新問題及思路詳解
這篇文章主要介紹了Pytorch實(shí)現(xiàn)網(wǎng)絡(luò)部分層的固定不進(jìn)行回傳更新,實(shí)現(xiàn)思路就是利用tensor的requires_grad，每一個(gè)tensor都有自己的requires_grad成員，值只能為True和False，具體內(nèi)容詳情跟隨小編一起看看吧
2021-08-08
python對輸出的奇數(shù)偶數(shù)排序?qū)嵗a
在本篇內(nèi)容里小編給大家整理的是一篇關(guān)于python對輸出的奇數(shù)偶數(shù)排序?qū)嵗a內(nèi)容，有興趣的朋友們可以參考下。
2020-12-12
python實(shí)現(xiàn)視頻讀取和轉(zhuǎn)化圖片
今天小編就為大家分享一篇python實(shí)現(xiàn)視頻讀取和轉(zhuǎn)化圖片，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-12-12
python matplotlib實(shí)現(xiàn)將圖例放在圖外
這篇文章主要介紹了python matplotlib實(shí)現(xiàn)將圖例放在圖外，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
使用Playwright進(jìn)行視覺回歸測試詳解
這篇文章主要介紹了使用Playwright進(jìn)行視覺回歸測試詳解,視覺回歸測試是一種軟件測試技術(shù)，專注于檢測Web應(yīng)用程序或網(wǎng)站的用戶界面中的視覺變化和差異,需要的朋友可以參考下
2023-08-08
python tiktoken的簡介、安裝、使用方法
tiktoken是OpenAI于近期開源的Python第三方模塊,該模塊主要實(shí)現(xiàn)了tokenizer的BPE（Byte pair encoding）算法,并對運(yùn)行性能做了極大的優(yōu)化,本文將介紹python tiktoken的簡介、安裝、使用方法,感興趣的朋友跟隨小編一起看看吧
2023-10-10
利用Python實(shí)現(xiàn)顏色色值轉(zhuǎn)換的小工具
最近一個(gè)朋友說已經(jīng)轉(zhuǎn)用Zeplin很久了。Zeplin的設(shè)計(jì)稿展示頁面的顏色色值使用十進(jìn)制的 RGB 表示的，在 Android 中的顏色表示大多情況下都需要十六進(jìn)制的 RGB 表示。所以想寫個(gè)工作，當(dāng)輸入十進(jìn)制的RGB ，得到十六進(jìn)制的色值，最好可以方便復(fù)制。下面來一起看看吧。
2016-10-10
Python入門之后再看點(diǎn)什么好?
看完了基礎(chǔ)書，甚至看兩遍了，但自己寫的時(shí)候還是沒思路，我該怎么辦？這篇文章主要介紹了Python入門之后再看點(diǎn)什么好，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2018-03-03
python3.4用循環(huán)往mysql5.7中寫數(shù)據(jù)并輸出的實(shí)現(xiàn)方法
下面小編就為大家?guī)硪黄猵ython3.4用循環(huán)往mysql5.7中寫數(shù)據(jù)并輸出的實(shí)現(xiàn)方法。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2017-06-06