文件名称: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)