mindspore.runtime.set_memory

mindspore.runtime.set_memory(init_size='2GB', increase_size='2GB', max_size='1024GB', optimize_level='O0')

设置使用内存池实现的运行时设备内存管理的内存参数。

框架默认设置所有参数，如下所示。

参数：

• init_size (str) - 内存池初始大小。格式为 xxGB，默认为 2G。

• increase_size (str) - 内存池的大小增加。当当前内存池没有 足够的内存的时候，内存池将按此值扩展。格式为 xxGB，默认为 2G。

• max_size (str) - 内存池可用的最大内存。 实际使用的内存大小是设备和最大设备内存的可用内存的最小值。 格式为 xxGB, 默认为设备的最大可用内存，表示为 1024G。

• optimize_level (str) - 内存优化级别。该值必须在 [O0, O1] 中，默认值： O0。

样例：

>>> import mindspore as ms
>>> ms.set_device("Ascend", 1)
>>> ms.runtime.set_memory("10G", "2G", "60G", "O1")