将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
2022-01-23 20:03:51
3KB
Python
1