【昇腾】训练过程中功率监测，和英伟达4090对比

昇腾NPU功率监测

参考：https://www.hiascend.com/document/detail/zh/Atlas%20200I%20A2/24.1.RC2/re/npu/npusmi_023.html

npu-smi info -t power -i id

Nvidia GPU功率监测

nvidia-smi --query-gpu=power.draw --format=csv,noheader,nounits -i id

昇腾功率监测脚本

import subprocess
import time
import pandas as pd
import threading

# 初始化一个列表来存储功率记录及其时间戳
power_records = []

def get_power(npu_id):
    "获取指定NPU的功率"
    try:
        # 执行npu-smi info命令并捕获输出
        output = subprocess.check_output(['npu-smi', 'info', '-t', 'power', '-i', str(npu_id)]).decode('utf-8')
        # 解析输出中的功率信息
        for line in output.split('\n'):
            if 'NPU Real-time Power(W)' in line:
                power = line.split(':')[1].strip()
                return float(power)
    except subprocess.CalledProcessError as e:
        print(f"Failed to get power info: {
     
     e}")
    return None

def record_power(npu_id, interval):
    "记录指定时长内的功率信息"
    while True:
        power = get_power(npu_id)