解决pytorch报错BrokenPipeError: [Errno 32] Broken pipe

问题描述

BrokenPipeError: [Errno 32] Broken pipe

原因分析

Windows使用DataLoader时设置num_workers的问题。

解决方案

两种解决方案,任选一种即可!

方案1

把代码放到if __name__ == "__main__":下运行

if __name__ == "__main__":
    train_dataloader = DataLoader(
        dataset=dataset,
        batch_size=32,
        shuffle=True,
        num_workers=8
    )

方案2

num_workers不要更改,设置为默认值0。

温馨提示

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)

num_workers参数允许输入的数据类型是整型,表示加载数据的子进程数。也就是使用多进程来加载数据,这样效率更高一些。如果设置为0表示子进程数是0,也就是只用一个主进程来加载数据。数据量不多的情况下,不需要用多进程,因为进程的创建和销毁也花时间。

引用参考

https://blog.csdn.net/qq_33666011/article/details/81873217
https://pytorch.org/docs/stable/_modules/torch/utils/data/dataloader.html
  • 16
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
当在使用yolov5s时出现了"BrokenPipeError: [Errno 32] Broken pipe"错误,这是一个Broken Pipe错误,在Python中常见的错误之一。该错误通常是由于子进程在主进程完成引导阶段之前尝试启动而引起的。为了解决这个问题,可以按照以下步骤进行操作: 1. 在主模块中使用适当的惯用语法,即`if __name__ == '__main__':`,这样可以确保子进程在正确的时机启动。可以在主模块的开头加上这样一行代码,以确保在主模块被执行时,才会启动子进程。 2. 如果程序不需要冻结以生成可执行文件,可以省略"freeze_support()"语句。 3. 另外,还可以尝试将使用torch.utils.data.DataLoader函数时的参数num_workers设置为0。这个参数指定了在数据加载过程中使用的子进程数量。将其设置为0可以避免在多进程加载数据时可能出现的问题。 综上所述,为了解决yolov5s报错"BrokenPipeError: [Errno 32] Broken pipe",可以尝试使用以上方法来修复该错误。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [Pytorch cifar10_tutorial.py问题BrokenPipeError: [Errno 32] Broken pipe【附完整代码】](https://blog.csdn.net/rllrllrll/article/details/88037353)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [Python提示[Errno 32]Broken pipe导致线程crash错误解决方法](https://download.csdn.net/download/weixin_38519849/13772038)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Xavier Jiezou

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值