python如何限制CPU和內存使用量

這篇文章主要介紹了python如何限制CPU和內存使用量，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

限制「CPU」和內存使用量

如果不是想優化程序對內存或 CPU 的使用率，而是想直接將其限制為某個確定的數字，Python 也有一個對應的庫可以做到：

import signal  
import resource  
import os  
# To Limit CPU time  
def time_exceeded(signo, frame):  
    print("CPU exceeded...")  
    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)  
# To limit memory usage  
def set_max_memory(size):  
    soft, hard = resource.getrlimit(resource.RLIMIT_AS)  
    resource.setrlimit(resource.RLIMIT_AS, (size, hard))

我們可以看到，在上面的代碼片段中，同時包含設置最大 CPU 運行時間和最大內存使用限制的選項。在限制 CPU 的運行時間時，我們首先獲得該特定資源（RLIMIT_CPU）的軟限制和硬限制，然后使用通過參數指定的秒數和先前檢索到的硬限制來進行設置。最后，如果 CPU 的運行時間超過了限制，我們將發出系統退出的信號。在內存使用方面，我們再次檢索軟限制和硬限制，并使用帶「size」參數的「setrlimit」和先前檢索到的硬限制來設置它。

感謝你能夠認真閱讀完這篇文章，希望小編分享的“python如何限制CPU和內存使用量”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!