这里写自定义目录标题
PyTorch多进程子进程瘫痪
你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Markdown编辑器, 可以仔细阅读这篇文章,了解一下Markdown的基本语法知识。
问题描述
当使用torch.multiprocessing module运行程序,由于子进程瘫痪,process.join()会卡住,导致主进程不继续执行。
原代码
if __name__ == '__main__':
for i in range(Max_Processes_p1):
p = mp.Process(target=do_something, args=(i, a))
p