BrokenPipeError: [Errno 32] Broken pipe解决方法

条件反射104

于 2020-05-19 17:33:37 发布

阅读量1w

点赞数 5

CC 4.0 BY-SA版权

分类专栏： anaconda+pycharm+pip等

本文链接：https://blog.csdn.net/qq_40317204/article/details/106219275

anaconda+pycharm+pip等专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种常见的PyTorch DataLoader运行时错误——BrokenPipeError，并提供了将num_workers参数设置为0的解决方案，以避免多线程数据加载时可能出现的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用pytorch运行代码报错：

BrokenPipeError: [Errno 32] Broken pipe

解决方案：

令 torch.utils.data.DataLoader() 函数的 num_workers = 0

例如：

train_loader = torch.utils.data.DataLoader(
    SVHNDataset(train_path, train_label,
                transforms.Compose([
                    transforms.Resize((64, 128)),
                    transforms.RandomCrop((60, 120)),
                    transforms.ColorJitter(0.3, 0.3, 0.2),
                    transforms.RandomRotation(5),
                    transforms.ToTensor(),
                    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
    ])), 
    batch_size=40, 
    shuffle=True, 
    num_workers=0, # 在此处，把num_workers设为0
)