我正在计算Python中编写的tensorRT脚本的功耗。CUDA中有nvprof cuda_script
,但我的Python脚本中找不到类似的值。
Python也有类似的东西吗?人们是如何计算p100/v100的功耗的?
类似于Nvidia GPU的How to profile CPU usage of a Python script?
我正在计算Python中编写的tensorRT脚本的功耗。CUDA中有nvprof cuda_script
,但我的Python脚本中找不到类似的值。
Python也有类似的东西吗?人们是如何计算p100/v100的功耗的?
类似于Nvidia GPU的How to profile CPU usage of a Python script?
2条答案
按热度按时间cu6pst1q1#
事实证明,在一个python文件中以线程方式运行这个进程是非常复杂的。解决方案是与bash同时运行它们,然后让power脚本检查另一个进程是否正在运行。示例代码如下:
nfg76nw02#
您可以使用eco2AI库来估计功耗。https://github.com/sb-ai-lab/Eco2AI
它被设计用于跟踪ML模型训练期间的二氧化碳排放量,但它也分别测量GPU、CPU和RAM的功耗。