python 使用 pynvml 工具获取GPU工作状态

╰+哭是因爲堅強的太久メ 2022-12-19 11:38 97阅读 0赞

> 深度学习过程中经常需要和GPU打交道，需要在训练之前确定GPU工作状态。本文介绍python中的pynvml包的使用方法，用来获取上述信息。

### 安装 ###

> 直接通过pip安装：

pip install nvidia-ml-py

> 或者根据所使用的python版本安装对应包：

pip install nvidia-ml-py2  # python2
    pip install nvidia-ml-py3  # python3

### 使用方法 ###

# 导入包
    import pynvml 
    # 每MB包含的字节数
    NUM_EXPAND = 1024 * 1024
    # 初始化
    pynvml.nvmlInit()

#GPU驱动信息
    pynvml.nvmlSystemGetDriverVersion()
    -> b'441.22'

#获取Nvidia GPU块数
    pynvml.nvmlDeviceGetCount()
    -> 1

#获取GPU 0 的handle
    handle = pynvml.nvmlDeviceGetHandleByIndex(0)

# 设备名称
    pynvml.nvmlDeviceGetName(handle)
    -> b'GeForce GTX 1660'

#通过handle获取GPU 0 的信息
    info = pynvml.nvmlDeviceGetMemoryInfo(handle)

#GPU 0 的总显存
    info.total 
    -> 6442450944
    info.total / NUM_EXPAND
    -> 6144.0

#GPU 0 使用的显存
    info.used
    -> 383299584
    info.used / NUM_EXPAND
    -> 365.54296875

#关闭 pynvml
    pynvml.nvmlShutdown()

### 踩坑 ###

*  报错 ： pynvml.NVMLError\_LibraryNotFound: NVML Shared Library Not Found

> 坑已填：[解决 pynvml 报错 NVML Shared Library Not Found / WinError 126][pynvml _ NVML Shared Library Not Found _ WinError 126]

### 参考资料 ###

*  http://www.ngui.cc/51cto/show-33630.html

[pynvml _ NVML Shared Library Not Found _ WinError 126]: https://www.zywvvd.com/2020/11/06/deep_learning/pynvml_model_not_found/pynvml_model_not_found/

发表评论取消回复

表情：

评论列表（有 0 条评论，97人围观）

还没有评论，来说两句吧...

相关阅读

相关 python 使用urllib3获取请求状态

要使用Python的urllib3库来获取请求状态，需要先安装urllib3库。可以使用以下命令在终端中安装该库： pip install urllib3 安装完成

布满荆棘的人生/ 2023年10月13日 12:54/ 0 赞/ 43 阅读

相关 python 使用 pynvml 工具获取GPU工作状态

> 深度学习过程中经常需要和GPU打交道，需要在训练之前确定GPU工作状态。本文介绍python中的pynvml包的使用方法，用来获取上述信息。安装 > 直接通过pi

╰+哭是因爲堅強的太久メ/ 2022年12月19日 11:38/ 0 赞/ 98 阅读

相关 Python获取CPU使用率、内存使用率、网络使用状态

[Python获取CPU使用率、内存使用率、网络使用状态 ][Python_CPU_] 分类： [ Python][Python] 2012-12-05 08:45 6

淡淡的烟草味﹌/ 2022年08月06日 06:10/ 0 赞/ 233 阅读

相关 Adreno GPU Profiler工具使用总结

http://blog.csdn.net/daijy0111/article/details/50427758 介绍 Adreno Profiler 是高通公司开发的一

素颜马尾好姑娘i/ 2022年06月13日 06:23/ 0 赞/ 361 阅读

相关 CPU GPU设计工作原理（转载）

原文地址：[http://www.cnblogs.com/geniusalex/archive/2008/12/26/1941766.html][http_www.cnblog

川长思鸟来/ 2022年06月10日 07:09/ 0 赞/ 229 阅读

相关 GPU如何工作

http://www.cnblogs.com/twjcnblog/archive/2012/09/19/2682911.html 早在1990年，无处不在的交

￡神魔★判官ぃ/ 2022年06月10日 02:16/ 0 赞/ 245 阅读

相关 GPU是如何工作的？

http://www.sohu.com/a/111539041\_472927 GPU是显示卡的“心脏”，也就相当于CPU在电脑中的作用，它决定了该显卡的档次和大部

一时失言乱红尘/ 2022年06月10日 02:16/ 0 赞/ 272 阅读

相关 sigar获取机器的使用状态

[Sigar介绍与使用][Sigar] 博客分类： [开源软件][Link 1] Sigar是Hyperic-hq产品的基础包,是Hyperi

偏执的太偏执、/ 2022年06月08日 08:14/ 0 赞/ 113 阅读

相关 Linux下实时查看GPU状态

1. 显示当前GPU使用情况 Nvidia自带了一个nvidia-smi的命令行工具，会显示显存使用情况： $ nvidia-smi 输出：

忘是亡心i/ 2022年04月08日 12:12/ 0 赞/ 343 阅读

相关 GPU工作原理与理解

本周看GPU看得有点儿乱， GPU英文全称Graphic Processing Unit，中文翻译为“[图形处理器][Link 1]”。由于GPU

快来打我*/ 2021年09月10日 09:06/ 0 赞/ 382 阅读