[英]Limit RAM usage to python program
我试图将Python程序的RAM使用量限制为一半,以便在使用所有RAM时都不会完全冻结,为此,我正在使用以下代码,该代码无法正常工作,并且笔记本电脑仍在冻结:
import sys
import resource
def memory_limit():
rsrc = resource.RLIMIT_DATA
soft, hard = resource.getrlimit(rsrc)
soft /= 2
resource.setrlimit(rsrc, (soft, hard))
if __name__ == '__main__':
memory_limit() # Limitates maximun memory usage to half
try:
main()
except MemoryError:
sys.stderr.write('MAXIMUM MEMORY EXCEEDED')
sys.exit(-1)
我正在使用从main
函数调用的其他函数。
我究竟做错了什么?
提前致谢。
PD:我已经搜索了一下,找到了我输入的代码,但是仍然无法正常工作...
好的,所以我进行了一些研究,找到了一个从Linux系统获取内存的函数: 在Python中确定可用RAM,然后我对其进行了一些修改,以获取可用内存并将最大可用内存设置为一半。
码:
def memory_limit():
soft, hard = resource.getrlimit(resource.RLIMIT_AS)
resource.setrlimit(resource.RLIMIT_AS, (get_memory() * 1024 / 2, hard))
def get_memory():
with open('/proc/meminfo', 'r') as mem:
free_memory = 0
for i in mem:
sline = i.split()
if str(sline[0]) in ('MemFree:', 'Buffers:', 'Cached:'):
free_memory += int(sline[1])
return free_memory
if __name__ == '__main__':
memory_limit() # Limitates maximun memory usage to half
try:
main()
except MemoryError:
sys.stderr.write('\n\nERROR: Memory Exception\n')
sys.exit(1)
将其设置为一半的行是resource.setrlimit(resource.RLIMIT_AS, (get_memory() * 1024 / 2, hard))
,其中get_memory() * 1024 / 2
将其设置为一半(以字节为单位)。
希望以后可以在同一件事上帮助他人! =)
最好使用:
if str(sline[0]) == 'MemAvailable:':
free_memory = int(sline[1])
break
答案代码在具有1TB RAM的计算机上为我提供了8 GB可用内存
我修改了@Ulises CT的答案。 因为我认为更改太多原始功能不是很好,所以我将其转为装饰器。 希望对您有所帮助。
import resource
import platform
import sys
def memory_limit(percentage: float):
"""
只在linux操作系统起作用
"""
if platform.system() != "Linux":
print('Only works on linux!')
return
soft, hard = resource.getrlimit(resource.RLIMIT_AS)
resource.setrlimit(resource.RLIMIT_AS, (get_memory() * 1024 * percentage, hard))
def get_memory():
with open('/proc/meminfo', 'r') as mem:
free_memory = 0
for i in mem:
sline = i.split()
if str(sline[0]) in ('MemFree:', 'Buffers:', 'Cached:'):
free_memory += int(sline[1])
return free_memory
def memory(percentage=0.8):
def decorator(function):
def wrapper(*args, **kwargs):
memory_limit(percentage)
try:
function(*args, **kwargs)
except MemoryError:
mem = get_memory() / 1024 /1024
print('Remain: %.2f GB' % mem)
sys.stderr.write('\n\nERROR: Memory Exception\n')
sys.exit(1)
return wrapper
return decorator
@memory(percentage=0.8)
def main():
print('My memory is limited to 80%.')
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.