昇思MindSpore进阶教程--内存复用

大家好，我是刘明，明志科技创始人，华为昇思MindSpore布道师。
技术上主攻前端开发、鸿蒙开发和AI算法研究。
努力为大家带来持续的技术分享，如果你也喜欢我的文章，就点个关注吧

正文开始

内存复用功能(Mem Reuse)是让不同的Tensor共用同样的一部分内存，以降低内存开销，支撑更大的网络，关闭后每个Tensor有自己独立的内存空间，Tensor间无共享内存。 MindSpore内存复用功能默认开启，可以通过以下方式手动控制该功能的关闭和开启。

使用方法

1. 创建配置文件mindspore_config.json。

{
    "sys": {
        "mem_reuse": true
    }
}

2. 通过 context 配置内存复用功能。

import mindspore as ms
ms.set_context(env_config_path="./mindspore_config.json")