mindspore.dataset.config.set_enable_shared_mem
- mindspore.dataset.config.set_enable_shared_mem(enable)
Set whether to use shared memory for interprocess communication when data processing multiprocessing is turned on.
Using shared memory can speed up the efficiency of data transfer between processes.
Shared memory is used by default.
Note
Windows and MacOS systems are not supported yet.
- Parameters
enable (bool) – Whether to use shared memory for interprocess communication.
- Raises
TypeError – If enable is not of type bool.
Examples
>>> # Enable shared memory feature to improve the performance of Python multiprocessing. >>> import mindspore.dataset as ds >>> ds.config.set_enable_shared_mem(True)