python如何给内存和cpu使用量设置限制
更新时间:2022年05月16日 10:33:59 作者:小平爱吃肉
这篇文章主要介绍了python如何给内存和cpu使用量设置限制,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
给内存和cpu使用量设置限制
在linux系统中,使用Python对内存和cpu使用量设置限制需要通过resource模块来完成。
resource文档地址:resource — Resource usage information
限制Python进程cpu使用时间的样例如下
import signal import resource import os def time_exceeded(signo, frame): print("time's up") raise SystemExit(1) def set_max_runtime(seconds): 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(5) while True: pass
运行上述代码,当超时时会产生SIGXCPU信号。程序就会做清理工作然后退出。
要限制内存的使用可以使用如下函数
def limit_memory(maxsize): soft, hard = resource.getrlimit(resource.RLIMIT_AS) resource.setrlimit(resource.RLIMIT_AS, (maxsize, hard))
当设定了内存限制后,如果没有更多的内存可用,程序就会开始产生MemoryError异常。
注:以上示例代码来源于:《Python Cookbook》P575 “给内存和cpu使用量设置限制”。
查询windows的cpu、内存使用率
# -*- coding: UTF-8 -*- import os def get_info(metric): metric_cmd_map = { "cpu_usage_rate": "wmic cpu get loadpercentage", "mem_total": "wmic ComputerSystem get TotalPhysicalMemory", "mem_free": "wmic OS get FreePhysicalMemory" } out = os.popen("{}".format(metric_cmd_map.get(metric))) value = out.read().split("\n")[2] out.close() return float(value) # cpu使用率 cpu_usage_rate = get_info('cpu_usage_rate') print("windows的CPU使用率是{}%".format(cpu_usage_rate)) # 无法直接查出内存使用率,总内存单位是b,而剩余内存单位是kb mem_total = get_info('mem_total')/1024 mem_free = get_info('mem_free') mem_usage_rate = (1 - mem_free/mem_total)*100 print("windows的内存使用率是{}%".format(mem_usage_rate))
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python pandas按行、按列遍历DataFrame的几种方式
在python的DataFrame中,因为数据中可以有多个行和列,而且每行代表一个数据样本,我们可以将DataFrame看作数据表,那你知道如何按照数据表中的行遍历吗,下面这篇文章主要给大家介绍了关于Python pandas按行、按列遍历DataFrame的几种方式,需要的朋友可以参考下2022-09-09keras tensorflow 实现在python下多进程运行
今天小编就为大家分享一篇keras tensorflow 实现在python下多进程运行,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
最新评论