python 查看程序的GPU显存占用
主要针对显卡：nvidia初衷：想要看某个python程序的GPU显存占用量在linux(1)nvidia-smi; (2)任务管理器 (3)pynvml库
·
显示GPU显存占用方法
引言
主要针对显卡：nvidia
初衷：想要看某个python程序的GPU显存占用量
一、nvidia-smi
在linux下使用nvidia-smi 可以直接显示GPU使用情况1:
但是在windows下有的时候显示N/A (如下图所示)：
二、windows下的任务管理器
任务管理器也可以查看程序GPU占用信息, 点击详细信息, 看专用GPU
三、pynvml库
比较全4:python 输出 GPU内存 最大使用率
简洁仅显存5:python查看显卡gpu信息
只有输出总GPU的显存占用，并没有找到获取单个程序GPU显存占用的函数
计算单个程序占用会有误差(桌面等程序也会占用GPU,占用量是动态变化的)
四、显存不够用又没钱怎么办
减小batch、减小网络、多用pooling降低数据分辨率 – 可能会降低性能
小知识6:Pytorch节省显存小技巧
希望可以做一个不费内存速度快精度还高的实用模型 
昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)