mindspore.dataset.config.set_enable_shared_mem
- mindspore.dataset.config.set_enable_shared_mem(enable)
设置共享内存标志的是否启用。如果 shared_mem_enable 为
True
,则使用共享内存队列将数据传递给为数据集操作而创建的进程,而这些数据集操作将设置 python_multiprocessing 为True
。说明
Windows和MacOS平台尚不支持 set_enable_shared_mem 。
- 参数:
enable (bool) - 表示当 python_multiprocessing 为
True
时,是否在数据集操作中使用共享内存。
- 异常:
TypeError - enable 不是bool类型。
样例:
>>> # Enable shared memory feature to improve the performance of Python multiprocessing. >>> import mindspore.dataset as ds >>> ds.config.set_enable_shared_mem(True)