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