335 Star 1.5K Fork 861

MindSpore / docs

Create your Gitee Account
Explore and code with more than 12 million developers,Free private repositories !:)
Sign up
Clone or Download
mem_reuse.md 1.03 KB
Copy Edit Raw Blame History
TingWang authored 2023-09-18 10:08 . update link logo

内存复用

查看源文件

概述

内存复用功能(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
master

Search