파이썬 메모리 체크

출처 : python 프로그램 메모리 사용량 확인

 def memory_usage(message: str = 'debug'):

    # current process RAM usage
    p = psutil.Process()
    rss = p.memory_info().rss / 2 ** 20 # Bytes to MB
    print(f"[{message}] memory usage: {rss: 10.5f} MB")

아래처럼 동작시킬 수 있겠네요.

memory_usage('#1')

# TODO
work_1()
memory_usage('#2')
work_2()

memory_usage('#3')
물리 메모리 사용량은 VmRSS를 보시면 됩니다>>> import psutil
>>> p = psutil.Process()
>>> p.memory_info()
pmem(rss=15491072, vms=84025344, shared=5206016, text=2555904, lib=0, data=9891840, dirty=0)