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捕捉和模拟鼠标事件的方法

    Python捕捉和模拟鼠标事件的方法

    这篇文章主要介绍了Python捕捉和模拟鼠标事件的方法,涉及PyHook和PyWin32模块的使用技巧,需要的朋友可以参考下
    2015-06-06
  • 对pyqt5多线程正确的开启姿势详解

    对pyqt5多线程正确的开启姿势详解

    今天小编就为大家分享一篇对pyqt5多线程正确的开启姿势详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • 浅谈Tensorflow模型的保存与恢复加载

    浅谈Tensorflow模型的保存与恢复加载

    本篇文章主要介绍了浅谈Tensorflow模型的保存与恢复加载,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-04-04
  • Python Tornado 实现SSE服务端主动推送方案

    Python Tornado 实现SSE服务端主动推送方案

    SSE是Server-Sent Events 的简称,是一种服务器端到客户端(浏览器)的单项消息推送,本文主要探索两个方面的实践一个是客户端发送请求,服务端的返回是分多次进行传输的,直到传输完成,这种情况下请求结束后,考虑关闭SSE,所以这种连接可以认为是暂时的,感兴趣的朋友一起看看吧
    2024-01-01
  • 利用python打印出菱形、三角形以及矩形的方法实例

    利用python打印出菱形、三角形以及矩形的方法实例

    最近在开发中遇到一个问题,需要利用python实现菱形、三角形以及矩形等形状,发现网上这方面的资料较少,所以总结分享下,这篇文章主要给大家介绍了关于利用python打印出菱形、三角形以及矩形的相关资料,需要的朋友可以参考借鉴,下面来一起看看吧。
    2017-08-08
  • python之virtualenv的简单使用方法(必看篇)

    python之virtualenv的简单使用方法(必看篇)

    下面小编就为大家分享一python之virtualenv的简单使用方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2017-11-11
  • python实现简单石头剪刀布游戏

    python实现简单石头剪刀布游戏

    这篇文章主要介绍了python实现简单石头剪刀布游戏,相信大家在童年或者生活中都玩过石头剪刀布这个游戏,这个游戏需要两个及以上的人。而今天,网上也实现了石头剪刀布的游戏。通过初步学习python,也学会了如何编写这个游戏。下面一起来看看详细内容吧
    2021-10-10
  • pandas读取csv文件提示不存在的解决方法及原因分析

    pandas读取csv文件提示不存在的解决方法及原因分析

    这篇文章主要介绍了pandas读取csv文件提示不存在的解决方法及原因分析,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • 解决python 两个时间戳相减出现结果错误的问题

    解决python 两个时间戳相减出现结果错误的问题

    这篇文章主要介绍了解决python 两个时间戳相减出现结果错误的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-03-03
  • Django权限设置及验证方式

    Django权限设置及验证方式

    这篇文章主要介绍了Django权限设置及验证方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-05-05

最新评论