BrokenPipeError: [Errno 32] Broken pipe 2019-10-26

运行Pytorch tutorial代码报错:BrokenPipeError: [Errno 32] Broken pipe

源代码地址:https://github.com/fanfanda/Medical-Image-Fusion

该问题的产生是由于windows下多线程的问题,和DataLoader类有关,具体细节点这里Fix memory leak when using multiple workers on Windows

解决方案:

修改调用torch.utils.data.DataLoader()函数时的 num_workers 参数。该参数官方API解释如下:

|

  • num_workers (int, optional) – how many subprocesses to use for data loading. 0
    means that the data will be loaded in the main process. (default: 0)

|

该参数是指在进行数据集加载时,启用的线程数目。截止当前2018年5月9日11:15:52,如官方未解决该BUG,则可以通过修改num_works参数为 **0 **,只启用一个主进程加载数据集,避免在windows使用多线程即可。

参考:https://blog.csdn.net/qq_33666011/article/details/81873217

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容