jupyter-resource-usage：Jupyter 资源监控的实用扩展

jupyter-resource-usage：Jupyter 资源监控的实用扩展

用 Jupyter 跑大数据或训练模型时，内核突然死掉、内存爆掉的情况很多人都遇到过。问题往往出在不清楚当前资源消耗到了什么程度。jupyter-resource-usage 这个扩展就是来解决这个问题的，它在 JupyterLab 和 Notebook 的状态栏实时显示内存和 CPU 占用，每 5 秒刷新一次。

目前这个项目在 GitHub 上有 536 个 Star，属于 Jupyter 官方生态的一部分，维护状态稳定。

能看什么

这个扩展主要显示三类资源信息：

内存占用。这是最核心的功能，状态栏会显示当前 notebook 服务器及其子进程（内核、终端等）占用的内存。在 Linux 系统下优先显示 PSS（比例集大小），其他平台回退到 RSS。

CPU 使用率。需要手动开启，默认不启用。开启后状态栏会同时显示 CPU 利用率。

磁盘占用。可以追踪指定分区的使用情况，同样在配置中开启后才显示。

对于 IPython 内核，还能在右侧边栏查看更详细的内核级资源消耗。Notebook 7.x 用户可以通过 View -> Right Sidebar -> Show Kernel Usage 开启，JupyterLab 用户直接点右侧边栏的 tachometer 图标。

安装方法

JupyterLab 4.x 和 Notebook 7.x 用户直接装最新版：

pip install jupyter-resource-usage

或者用 conda：

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

JupyterLab 3.x 和 Notebook 6.x 用户需要装 1.0.0 以下的版本：

pip install 'jupyter-resource-usage<1.0.0'

几个实用配置

内存上限警告。可以设置一个内存上限值，当使用量接近上限时状态栏背景会变红提醒。阈值默认是上限的 90%，也就是剩余 10% 时触发警告。这个上限本身只是显示，不会真的限制内存使用。

隐藏主机信息。默认情况下 Kernel Usage 边栏会显示 Host CPU 和 Host Virtual Memory。如果你不想看到主机层面的信息，可以设置 show_host_usage=False。

关闭 Prometheus 指标。如果界面出现卡顿，可能是 Prometheus 指标的一个已知 bug 导致的。可以在启动参数里加上 --ResourceUseDisplay.enable_prometheus_metrics=False 来禁用。

替代前端。除了默认的状态栏样式，还可以切换成另一种顶部栏显示方式。需要在 Settings -> Settings Editor -> Resource Usage Indicator 里勾选启用，刷新页面后生效。

实际体验

这个扩展的定位很明确：让用户随时知道自己占了多少资源。对于经常跑数据处理任务的人来说，这能避免很多因为内存不足导致的意外中断。

536 个 Star 说明它解决的是真实需求，而非包装出来的伪需求。安装简单，配置选项也够用，属于那种装上后基本不用管、默默发挥作用的工具。

如果你每天都在用 Jupyter，装一个没什么负担。至少下次内核挂掉之前，你能提前从状态栏看到内存已经飙红了。

如果你每天都在用 Jupyter，装一个没什么负担。至少下次内核挂掉之前，你能提前从状态栏看到内存已经飙红了。