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 一维二维插值实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • Python pandas按行、按列遍历DataFrame的几种方式

    Python pandas按行、按列遍历DataFrame的几种方式

    在python的DataFrame中,因为数据中可以有多个行和列,而且每行代表一个数据样本,我们可以将DataFrame看作数据表,那你知道如何按照数据表中的行遍历吗,下面这篇文章主要给大家介绍了关于Python pandas按行、按列遍历DataFrame的几种方式,需要的朋友可以参考下
    2022-09-09
  • 利用 Monkey 命令操作屏幕快速滑动

    利用 Monkey 命令操作屏幕快速滑动

    Monkey测试是Android平台自动化测试的一种手段,通过Monkey程序模拟用户触摸屏幕、滑动Trackball、按键等操作来对设备上的程序进行压力测试,检测程序多久的时间会发生异常
    2016-12-12
  • Swift 3.0在集合类数据结构上的一些新变化总结

    Swift 3.0在集合类数据结构上的一些新变化总结

    与字符串类似,Swift3在数组和字典等结构上也有改动,这里我们就来整理Swift 3.0在集合类数据结构上的一些新变化总结:
    2016-07-07
  • Python与HTTP服务交互的三种方式

    Python与HTTP服务交互的三种方式

    本文主要介绍了Python与HTTP服务交互的三种方式,通过http.client,requests,RissionPage,具有一定的参考价值,感兴趣的可以了解一下
    2024-03-03
  • Python 如何实现文件自动去重

    Python 如何实现文件自动去重

    这篇文章主要介绍了Python 实现文件自动去重操作,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-06-06
  • 详解matplotlib技巧之缩放和投影

    详解matplotlib技巧之缩放和投影

    我们在使用matplotlib绘制图形时,有两个重要的技巧:一种是 Scale(缩放),一种是Projection(投影),缩放和投影在matplotlib绘图中起着至关重要的作用,它们帮助我们更好地展示和理解数据,本文将详细的给大家介绍这两个技巧,需要的朋友可以参考下
    2024-01-01
  • Python计算时间间隔(精确到微妙)的代码实例

    Python计算时间间隔(精确到微妙)的代码实例

    今天小编就为大家分享一篇关于Python计算时间间隔(精确到微妙)的代码实例,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-02-02
  • Python实现8种常用抽样方法

    Python实现8种常用抽样方法

    抽样是统计学、机器学习中非常重要,本文就用Python实现抽样方法,主要介绍了八种方法,感兴趣的小伙伴们可以参考一下
    2021-06-06
  • keras tensorflow 实现在python下多进程运行

    keras tensorflow 实现在python下多进程运行

    今天小编就为大家分享一篇keras tensorflow 实现在python下多进程运行,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02

最新评论