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))

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python3新增的Byte类型解读

    Python3新增的Byte类型解读

    这篇文章主要介绍了Python3新增的Byte类型,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-09-09
  • 利用Python实现QQ实时到账免签支付功能

    利用Python实现QQ实时到账免签支付功能

    这篇文章主要介绍了利用Python实现QQ实时到账免签支付功能,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-03-03
  • python自动化实现登录获取图片验证码功能

    python自动化实现登录获取图片验证码功能

    这篇文章主要介绍了python自动化实现登录获取图片验证码功能,本文通过实例截图的形式给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-11-11
  • Pandas中Apply函数加速百倍的技巧分享

    Pandas中Apply函数加速百倍的技巧分享

    虽然目前dask,cudf等包的出现,使得我们的数据处理大大得到了加速,但是并不是每个人都有比较好的gpu。本文为大家分享几个Pandas中Apply函数加速百倍的技巧,希望有所帮助
    2022-07-07
  • 从零学Python之入门(二)基本数据类型

    从零学Python之入门(二)基本数据类型

    这是继“hello world”之后的第二篇入门级基础知识,以后这个系列会按照入门、进阶、精通三个分类进行下去,欢迎高手们来拍砖
    2014-05-05
  • Python实现的双色球生成功能示例

    Python实现的双色球生成功能示例

    这篇文章主要介绍了Python实现的双色球生成功能,涉及Python基于random模块生成随机数的相关操作技巧,需要的朋友可以参考下
    2017-12-12
  • Python中dict排序的两种方法

    Python中dict排序的两种方法

    字典本身是无序的,所以它每次输出都是不一样的,顺序都是乱的,那么字典如何排序,本文主要介绍了Python中dict排序的两种方法,感兴趣的可以了解一下
    2024-01-01
  • python实现翻转棋游戏(othello)

    python实现翻转棋游戏(othello)

    这篇文章主要为大家详细介绍了python实现翻转棋游戏,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-07-07
  • 利用python下载scihub成文献为PDF操作

    利用python下载scihub成文献为PDF操作

    这篇文章主要介绍了利用python下载scihub成文献为PDF操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-07-07
  • Python如何实现爬取B站视频

    Python如何实现爬取B站视频

    这篇文章主要介绍了Python如何实现爬取B站视频,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-05-05

最新评论