Python調用pynvml查看GPU信息

安裝方法：pip install nvidia-ml-py

 

def checkGPU():
    import pynvml
    pynvml.nvmlInit()     #初始化
    print("Driver: ",pynvml.nvmlSystemGetDriverVersion())#顯示驅動信息
    deviceCount = pynvml.nvmlDeviceGetCount()
    for i in range(deviceCount):
        handle = pynvml.nvmlDeviceGetHandleByIndex(i)
        print("GPU", i, ":", pynvml.nvmlDeviceGetName(handle))
checkGPU()

 

————————————————
版權聲明：本文為CSDN博主「hitrjj」的原創文章，遵循 CC 4.0 BY-SA 版權協議，轉載請附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/u014636245/article/details/83932090