torch.multiprocessing

封装了multiprocessing模块。用于在相同数据的不同进程中共享视图。

一旦张量或者存储被移动到共享单元(见sharememory()),它可以不需要任何其他复制操作的发送到其他的进程中。

这个API与原始模型完全兼容,为了让张量通过队列或者其他机制共享,移动到内存中,我们可以

由原来的import multiprocessing改为import torch.multiprocessing。

由于API的相似性,我们没有记录这个软件包的大部分内容,我们建议您参考原始模块的非常好的文档。

warning: 如果主要的进程突然退出(例如,因为输入信号),Python中的multiprocessing有时会不能清理他的子节点。

这是一个已知的警告,所以如果您在中断解释器后看到任何资源泄漏,这可能意味着这刚刚发生在您身上。

results matching ""

    No results matching ""