Python查看GPU已使用的显存

TccccD

于 2018-06-22 15:36:40 发布

阅读量2.7w

点赞数 5

分类专栏： python 机器学习深度学习文章标签： python 深度学习 GPU 显存

本文链接：https://blog.youkuaiyun.com/u011094454/article/details/80774495

版权

机器学习同时被 3 个专栏收录

16 篇文章

订阅专栏

python

14 篇文章

订阅专栏

深度学习

3 篇文章

订阅专栏

本文介绍如何通过pip安装pynvml模块，并演示如何在Python环境中使用该模块初始化NVML库，获取指定GPU id的设备句柄，进而读取GPU的内存使用情况。

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

pip安装pynvml

pip install nvidia-ml-py3

如果是Python2，则pip nvidia-ml-py2

之后可以在Python3 下使用了

import pynvml
pynvml.nvmlInit()
# 这里的0是GPU id
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.used)

GPU其他信息，可以查看API文档