Python在Unix系统中限制内存与CPU使用量的方法
发布时间：2026-10-14| 5人浏览过
本文介绍如何在Unix系统中使用Python的resource模块限制程序的内存和CPU使用量。通过设置软限制和硬限制，程序在超时或超内存时会抛出异常并退出。该方法适用于Linux等Unix系统，但需注意不同系统的兼容性差异。
问题背景
问题
针对在Unix系统上运行的程序，需要对其内存或CPU的使用量进行限制。
解决方案与实现
解决方案
利用Python的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在Unix系统中限制内存和CPU使用量的方法详情，更多关于Python限制内存和CPU的资料，请持续关注本站其他相关文章！
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
