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

安装 {#安装}

直接通过pip安装： 或者根据所使用的python版本安装对应包：

使用方法 {#使用方法}

踩坑 {#踩坑}

• 报错 ： pynvml.NVMLError_LibraryNotFound: NVML Shared Library Not Found

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

参考资料 {#参考资料}

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

文章链接：
https://www.zywvvd.com/notes/study/deep-learning/pynvml-gpu-info-get/pynvml-gpu-info-get/