Python查看电脑的GPU型号

最新推荐文章于 2025-05-28 10:09:54 发布

Learning_is_endless

最新推荐文章于 2025-05-28 10:09:54 发布

阅读量3.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习

本文链接：https://blog.csdn.net/zhangxiaoduoduo/article/details/118228371

深度学习专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用Python库pynvml来查询计算机中安装的NVIDIA GPU的数量及具体型号。通过安装pynvml并运行提供的代码示例，用户可以轻松获取到关于GPU的基本信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Step1:先安装pynvml

pip install pynvml

Step2:使用如下代码

from pynvml import *
nvmlInit()

deviceCount = nvmlDeviceGetCount()#几块显卡
print(deviceCount)

for i in range(deviceCount):
    handle = nvmlDeviceGetHandleByIndex(i)
    print ("Device", i, ":", nvmlDeviceGetName(handle)) #具体是什么显卡