将RAM的使用限制为python程序

一尘不染

将RAM的使用限制为python程序

python

我试图将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：我已经搜索了一下，找到了我输入的代码，但是仍然无法正常工作…

阅读 133

2020-12-20

共1个答案

一尘不染

好的，所以我进行了一些研究，找到了一个从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设置它的一半（它以字节为单位）。

希望这可以在将来帮助其他人！=）

2020-12-20