diff --git a/mem_profile.py b/mem_profile.py new file mode 100644 index 0000000..62d6433 --- /dev/null +++ b/mem_profile.py @@ -0,0 +1,10 @@ +import psutil +import os +import sys +import time + + +def memory_usage_psutil(): + process = psutil.Process(os.getpid()) + mem = process.memory_info().rss / float(2 ** 20) + return '{:.2f} MB'.format(mem)