Jupyter Notebookに現在のメモリ使用量を表示する

こんな疑問にお答えします。

Jupyter Notebook上にメモリの表示はできるの？

結論・・・出来ます！！

ネットで調べてみるとどうやらあるパッケージをインストールすることで実現できることがわかりました。

pip や conda でインストールするだけで簡単に実現できます。

具体的な利用例を次に紹介します。

jupyter-resource-usage をインストール

Jupyter Notebook上にメモリ利用量表示するには jupyter-resource-usage をインストールします。

pipの場合：

pip install jupyter-resource-usage

condaの場合：

conda install -c conda-forge jupyter-resource-usage

ノートバックのバージョンを知るにはターミナルで以下のコマンドを実行します。

$ jupyter notebook --version
6.0.3

拡張機能を手動で有効：

jupyter serverextension enable --py jupyter-resource-usage --sys-prefix
jupyter nbextension install --py jupyter-resource-usage --sys-prefix
jupyter nbextension enable --py jupyter-resource-usage --sys-prefix

Jupyter Notebookを実行

インストール後はターミナルでJupyter Notebookを実行するだけでOKです。

ノートブックにアクセスすると右上の方に利用中のメモリ量が表示されています。

また、更なるオプションとしてメモリの利用閾値や警告の閾値をオプションで設定できるようです。

–ResourceUseDisplay.mem_limit や –ResourceUseDisplay.mem_warning_threshold=0.1 といったオプションをJupyterコマンド実行時などに渡せばOKのようです。

詳しくは公式ドキュメントを参照してください。

参考：jupyter-resource-usage

まとめ