mindspore框架在内存管理这块是如何设计的?目前在模型训练时,npu-smi info 看到的不是实际占用的显存,而是显示的max-device-memory的显存。如何查看实际占用的显存,类似英伟达的显示某一段时间的峰值显存。(目前的现象是不管batch_size设置多大,npu-smi info 显示占用的显存基本差不多,(32G的基本都显示在30-31G, 64G的基本显示在62-63G)) 个人感觉是训练时直接申请了max-device-memory这么大的显存?
基于mindspore框架,训练模型时的显存显示问题
已解决



