美文网首页
2020-03-23 python中如何查看GPU显存占用

2020-03-23 python中如何查看GPU显存占用

作者: Hie_9e55 | 来源:发表于2020-03-23 19:30 被阅读0次

可以使用pynvml库来查看

pynvml的安装

pip install nvidia-ml-py3
(尝试了一下,conda install似乎行不通)

pynvml的使用

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

pynvml文档:https://docs.nvidia.com/deploy/nvml-api/index.html

相关文章

网友评论

      本文标题:2020-03-23 python中如何查看GPU显存占用

      本文链接:https://www.haomeiwen.com/subject/umhgyhtx.html