【Python】监控GPU温度

因为疫情卡在宿舍，无奈笔记本灰太多缺少清灰工具，电脑散热太拉，开着cad一天之后经常蓝屏：(

因此借鉴该网站资料：Python使用pynvml查看GPU信息，写了一份监控电脑gpu的温度的小程序，在此做为摸鱼记录。

官网文档文件：需要翻墙才能查阅

import pynvml
import time

while 1:
    pynvml.nvmlInit() # 初始化

    # 获取GPU i的handle，后续通过handle来处理
    handle = pynvml.nvmlDeviceGetHandleByIndex(0)
  
    gpuTemperature = pynvml.nvmlDeviceGetTemperature(handle, 0) #读取温度
    print("温度：", gpuTemperature, "摄氏度")
    time.sleep(5) # 每5s测一次温度

    if gpuTemperature>90:
        print("温度超过90了！")
        break

    #最后关闭管理工具
    pynvml.nvmlShutdown() 

说明

Nvidia的显卡提供了 NVML（英伟达显卡管理库）以及构建在其上的 nvidia-smi 显卡系统管理界面)，可以方便的查询显卡的信息和工作状况。

在python中同样可以利用pynvml库来实现显卡信息的获取，在调用该库时需提前下载好该库文件。