上传者: 42162171
|
上传时间: 2022-01-23 20:03:51
|
文件大小: 3KB
|
文件类型: -
将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