Messen (max) Speicherverbrauch mit IPython-wie Zeit, aber Memit

Question

Ich habe eine einfache Aufgabe: Neben der Zeit messen, die es dauert, um ein Stück Code in Python auszuführen, muss ich die Menge an Speicher einen bestimmten Chunk messen Code benötigt.

IPython hat ein schönes Programm namens timeit, die wie folgt funktioniert:

In [10]: timeit 3 + 3 
10000000 loops, best of 3: 24 ns per loop 

Was ich suche so etwas wie dieses:

In [10]: memit 3 + 3 
10000000 loops, best of 3: 303 bytes per loop 

mir bewusst bin, dass dies tut wahrscheinlich nicht kommen mit IPython eingebaut - aber ich mag die timeit - memit Analogie.

Answer 1

In der Tat, es ist bereits vorhanden, als Teil des prosaisch namens memory_profiler Paket:

In [2]: %memit np.zeros(1e7) 
maximum of 3: 76.402344 MB per loop 

Mehr Infos unter https://github.com/fabianp/memory_profiler#ipython-integration

Edit: Um dies zu nutzen, müssen Sie zuerst es als IPython zu laden Erweiterung:

%load_ext memory_profiler 

um IPython immer beim Start der memory_profiler Erweiterung laden, fügen sie die c.InteractiveShellApp.extensions Liste in Ihrem Profil der ipython_config.py:

$ grep -C2 c.InteractiveShellApp.extensions ~/.ipython/profile_default/ipython_config.py 
# A list of dotted module names of IPython extensions to load. 
# 
c.InteractiveShellApp.extensions = [ 
    'autoreload', 
    'memory_profiler',