参考 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