MindSpore运行模式与PyNative内存调优分析
既然这样,那就需要将内存池的大小改小,以便多些内存给其它场景使用。context.set_context(mempool_block_size=“30GB”, max_device_memory=“31GB”),通过max_device_memory接口,提供给内存池的设备内存可用大小为31GB,去掉persistent block的1GB,取min(mempool_block_size, max_device_memory - 1GB)值,那么实际生效的common block大小就是30GB。