如何在Visual Studio Code中监控Jupyter Notebook的资源使用情况？

一、专用扩展推荐 #

• Jupyter Resource Usage：这是VS Code中适配Jupyter的主流资源监控扩展，安装后会在状态栏实时显示当前内核的内存占用量，点击状态栏的数值可展开面板，查看CPU使用率、内存细分数据等，逻辑和Colab的资源展示类似。

二、内置设置调整 #

在VS Code设置面板中搜索jupyter.resourceUsage，开启以下配置即可启用基础资源监控：

{
  "jupyter.resourceUsage.enabled": true,
  "jupyter.resourceUsage.showMemoryGauge": true,
  "jupyter.resourceUsage.showCpuUsage": true
}

设置生效后，状态栏会直接显示CPU和内存的实时占用状态。

三、代码单元格手动查询 #

如果需要更灵活的资源数据，可通过Python代码直接获取系统或当前进程的资源信息：

1. 先安装依赖库：

!pip install psutil

2. 执行代码获取RAM、CPU、磁盘（ROM）信息：

import psutil
import os

# 当前Notebook进程的内存使用
current_process = psutil.Process(os.getpid())
mem_usage = current_process.memory_info().rss / (1024 ** 2)
print(f"当前进程内存占用: {mem_usage:.2f} MB")

# 系统CPU实时使用率
cpu_usage = psutil.cpu_percent(interval=1)
print(f"系统CPU使用率: {cpu_usage}%")

# 磁盘（ROM）使用情况（Windows系统可将路径改为'C:\\'）
disk_stats = psutil.disk_usage('/')
print(f"磁盘总容量: {disk_stats.total / (1024 ** 3):.2f} GB")
print(f"磁盘已使用: {disk_stats.used / (1024 ** 3):.2f} GB")
print(f"磁盘剩余空间: {disk_stats.free / (1024 ** 3):.2f} GB")

内容的提问来源于stack exchange，提问作者pouya