問題

你想對在Unix系統上面執行的程式設定記憶體或CPU的使用限制。

解決方案

resource 模組能同時執行這兩個任務。例如，要限制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 訊號在時間過期時被生成，然後執行清理並退出。

要限制記憶體使用，設定可使用的總記憶體值即可，如下：

import resource

def limit_memory(maxsize):
 soft,hard = resource.getrlimit(resource.RLIMIT_AS)
 resource.setrlimit(resource.RLIMIT_AS,(maxsize,hard))

像這樣設定了記憶體限制後，程式執行到沒有多餘記憶體時會丟擲 MemoryError 異常。

討論

在本節例子中，setrlimit() 函式被用來設定特定資源上面的軟限制和硬限制。 軟限制是一個值，當超過這個值的時候作業系統通常會發送一個訊號來限制或通知該程序。 硬限制是用來指定軟限制能設定的最大值。通常來講，這個由系統管理員通過設定系統級引數來決定。 儘管硬限制可以改小一點，但是最好不要使用使用者程序去修改。

setrlimit() 函式還能被用來設定子程序數量、開啟檔案數以及類似系統資源的限制。 更多詳情請參考 resource 模組的文件。

需要注意的是本節內容只能適用於Unix系統，並且不保證所有系統都能如期工作。 比如我們在測試的時候，它能在Linux上面正常執行，但是在OS X上卻不能。

以上就是Python限制記憶體和CPU的使用量的方法的詳細內容，更多關於Python限制記憶體和CPU的資料請關注我們其它相關文章！