335 Star 1.5K Fork 867

MindSpore / docs

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mem_reuse.md 1.03 KB
一键复制 编辑 原始数据 按行查看 历史
TingWang 提交于 2023-09-18 17:41 . update link logo for 2.1

内存复用

查看源文件

概述

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

使用方法

  1. 创建配置文件mindspore_config.json

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

    mem_reuse: 控制内存复用功能是否开启,当设置为true时,控制内存复用功能开启,为false时,内存复用功能关闭。

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

    import mindspore as ms
    ms.set_context(env_config_path="./mindspore_config.json")
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r2.1

搜索帮助