Multiprocessing package - torch.multiprocessing

参考    Multiprocessing package - torch.multiprocessing - 云+社区 - 腾讯云

 torch.multiprocessing是围绕本机多处理模块的包装器。它注册自定义约简器，使用共享内存在不同进程中提供对相同数据的共享视图。一旦张量/存储被移动到shared_memory(请参阅share_memory_())，就可以将它发送到其他进程，而不需要进行任何复制。

该API与原始模块100%兼容—只需更改导入多处理即可导入torch。多处理将所有张量通过队列发送或通过其他机制共享，移动到共享内存。由于api的相似性，我们不记录这个包的大部分内容，我们建议参考原始模块中非常好的文档。

Warning

如果主进程突然退出(例如，由于传入的信号)，Python的多处理有时无法清理其子进程。这是一个已知的警告，因此，如果你在中断解释器之后看到任何资源泄漏，这可能意味着这只是发生在你身上。

Strategy management

torch.multiprocessing.get_all_sharing_strategies()[source]

返回当前系统支持的一组共享策略。

torch.multiprocessing.get_sharing_strategy()[source]

返回共享CPU张量的当前策略。

torch.multiprocessing.set_sharing_strategy(new_st