2024-11-24 厉飞雨 阅读(3) 评论(0) 赞(0)

> 深度学习过程中经常需要和GPU打交道，需要在训练之前确定GPU工作状态。本文介绍python中的pynvml包的使用方法，用来获取上述信息。 ### 安装 {#安装} > 直接通过pip安装： > 或者根据所使用的python版本安装对应包： ### 使用方法 {#使用方法} ### 踩坑 {#踩坑} * 报错 ： pynvml.NVMLErro...