Create mem_profile.py

2017-01-06 14:51:01 -07:00
parent 5d14af3220
commit 937a90a44e
1 changed files with 10 additions and 0 deletions
@@ -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)