Create mem_profile.py

sulaiman1988 · Jan 6, 2017 · 937a90a · 937a90a
1 parent 5d14af3
commit 937a90a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/mem_profile.py 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)