zwPython.com研习社

 找回密码
 立即注册
搜索
热搜: 活动 交友 discuz
查看: 351|回复: 0

简单实用的GPU内存监控程序(附:zw版python源码)

[复制链接]

228

主题

235

帖子

5249

积分

超级版主

Rank: 8Rank: 8

积分
5249
发表于 2021-5-21 08:07:20 | 显示全部楼层 |阅读模式
简单实用的GPU内存监控程序(附:zw版python源码)

在对高频tick、分时等数据进行量化回测分析,以及运行AI人工智能模型时,
仅靠cpu的算力是不行的,必须借助GPU强大的并行加速运算。
GPU运算数据是基于显存数据,与电脑主机是两个独立监控的体系。
如果每个批次batch数据量过大,会引发显存溢出错误,如果数据量过小,又浪费了昂贵的GPU显存资源。
精确调整优化每个批次batch数据量的大小,属于GPU和AI编程的基本功。
通常,可以调用Nvidia公司的命令行工具程序:nvidia-smi
对程序、显存进行实时监控。


win10新版本的任务管理器,也与时共进,加入了GPU监控,不过有些过于简单。
process explorer是免费的增强版任务管理器,也提供了GPU监控功能。

虽然以上几种办法都不错,不过对于编程而言,还是基于模块库的版本更加灵活。
万能的python平台,模块库pynvml就可以实现GPU的监控功能。

pynvml模块库安装脚本如下:
pip install nvidia-ml-py3


全文请关注公众号或浏览:
zwPython.com,ziwang.com,TopQuant.vip网站
请大家注意zwPython.com,ziwang.com,TopQuant.vip网站,公众号的最新消息。

以下是python调用案例源码:

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
【极宽QQ群】
群号:1057170501 ,zwPython.com学习QQ群,面向py初学者和大众小白用户。
群号:124134140,TOP极宽金融量化,三千人超大QQ群,面向专业金融实盘用户。
群号:699266891,TFB极宽足球大数据QQ群
群号:790548107,FASTAI智能量化QQ群。
【微信公众号】
请大家关注TOP极宽量化公众号,大量原创Python量化技术
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|zwPython.com研习社

GMT+8, 2021-6-24 11:55 , Processed in 0.032893 second(s), 4 queries , Gzip On, MemCache On.

Powered by Discuz! X3.4

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表