在 Python 程序运行过程中，可以通过一些第三方库来实时监控内存（RAM）和显存（GPU 显存，如 NVIDIA GPU）的使用情况。常用的工具包括： - `psutil`：用于监控 CPU 和内存使用情况。 - `GPUtil` 或 `nvidia-smi`：用于监控 NVIDIA GPU 的使用情况。 下面是一个完整的示例程序，它会在每次循环中打印当前的内存和显存占用情况（如果存在 GPU）： ### 示例代码： ```python import psutil import GPUtil import time def get_memory_usage(): # 获取系统内存使用情况 mem = psutil.virtual_memory() print(f"内存使用: {mem.used / (1024 ** 3):.2f} GB / {mem.total / (1024 ** 3):.2f} GB") def get_gpu_usage(): # 获取所有 GPU 的使用情况 gpus = GPUtil.getGPUs() for gpu in gpus: print(f"GPU {gpu.id} 使用情况:") print(f" 名称: {gpu.name}") print(f" 显存使用: {gpu.memoryUsed} MB / {gpu.memoryTotal} MB") print(f" GPU 利用率: {gpu.load * 100:.1f}%") def monitor_system(interval=1, duration=10): """ 实时监控系统资源使用情况 :param interval: 每次检测间隔时间（秒） :param duration: 总共监控时间（秒） """ start_time = time.time() while time.time() - start_time < duration: print("\n" + "-"*40) get_memory_usage() if GPUtil.getGPUs(): get_gpu_usage() else: print("未检测到 GPU 设备") time.sleep(interval) if __name__ == "__main__": monitor_system(interval=1, duration=10) ``` ### 安装依赖： 你需要安装以下库才能运行上述代码： ```bash pip install psutil gputil ``` --- ### 解释： - `psutil.virtual_memory()`：获取系统的虚拟内存信息，包括总内存、已用内存等。 - `GPUtil.getGPUs()`：返回一个包含所有 GPU 信息的对象列表，每个对象都有 `memoryUsed`, `memoryTotal`, `load` 等属性。 - `monitor_system()`：主函数，每隔 `interval` 秒检测一次系统资源，持续 `duration` 秒。 --- ### 示例输出（可能类似）： ``` ---------------------------------------- 内存使用: 3.25 GB / 16.00 GB GPU 0 使用情况: 名称: NVIDIA GeForce RTX 3080 显存使用: 1024 MB / 10240 MB GPU 利用率: 15.5% ``` ---