gpulog:使用nvidia-smi记录和显示GPU利用率的示例脚本

时间:2024-03-09 00:36:28
【文件属性】:

文件名称:gpulog:使用nvidia-smi记录和显示GPU利用率的示例脚本

文件大小:3KB

文件格式:ZIP

更新时间:2024-03-09 00:36:28

Python

将NVIDIA GPU利用率记录到文件的示例 此存储库包含一些小代码示例,这些示例说明如何使用nvidia-smi将GPU使用率记录到CSV文件中,以及如何使用python脚本来绘制结果。 使用脚本log_gpu_utilization.sh开始记录gpu利用率。 通过按CTRL+C退出日志记录。 要显示GPU利用率,请运行脚本plot_nvidia_dump.py : $ python plot_nvidia_dump.py gpu_log_[timestamp].csv 您可以通过提供GPU的整数ID来过滤掉GPU。 为了通过这些ID过滤器,可使用--filter-ids命令行参数plot_nvidia_dump.py : $ python plot_nvidia_dump.py gpu_log_[timestamp].csv --filter-ids 6 仅显示ID为6的GPU


【文件预览】:
gpulog-main
----README.md(1KB)
----plot_nvidia_dump.py(2KB)
----log_gpu_utilization.sh(146B)
----live_gpu_utlization_plot.py(3KB)

网友评论