python系统问题

本文介绍如何使用Python的resource模块来限制程序的最大内存占用,避免因内存溢出导致Docker容器崩溃。通过设置内存使用上限,当达到指定阈值时程序会抛出MemoryError。
部署运行你感兴趣的模型镜像

https://mp.weixin.qq.com/s/Bi7f6VwT8DoXbuozlX3PIw
限制程序最大内存使用

如果程序开发不当,可能会出现占用过多内存的情况。特别是在Docker里面,如果Python程序占用太多内存,可能会导致Docker容器死掉。

为了限制Python程序所能使用的最大内存,我们可以使用Python自带的resource模块。

首先获取系统默认的内存soft上限和hard上限:

import resource
soft, hard = resource.getrlimit(resource.RLIMIT_AS)

其中的resource.RLIMIT_AS实际上就是数字5,表示内存资源。而soft限制是一个可以调节的内存使用上限,hard是一个一旦设定就很难动态调整的内存使用量上限。

在我的电脑上运行效果如下图所示:

在这里插入图片描述
一开始soft和hard的值是一样的。并且它们的单位不是比特。而是比比特还小的单位。我的电脑是8GB内存,而8 * 1024 ^ 6 正好约等于这里的hard和soft的值。

接下来,设置当前运行的这个Python程序能够使用的最大内存:
resource.setrlimit(resource.RLIMIT_AS, (最大上限, hard))
其中最大上限是一个整数。假设我想限制当前程序最多可以使用500MB的内存,那么这里的最大上限可以设置为:
0.5 * 1024 ^ 6 = 576460752303423488
所以设置为:
resource.setrlimit(resource.RLIMIT_AS, (576460752303423488, hard))
那么当前Python程序使用的内存超过500MB以后,程序就会抛出MemoryError。

最后,这些代码只能在Linux系统上正常工作。Windows和Mac上可能会有问题。

您可能感兴趣的与本文相关的镜像

Python3.11

Python3.11

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值