快盘下载:好资源、好软件、快快下载吧!

快盘排行|快盘最新

当前位置:首页软件教程电脑软件教程 → win11系统pytorch下使用DataLoader加载数据,运行images,-labels-=-next(iter(train_loader))报错BrokenPipeError:-[Errno

win11系统pytorch下使用DataLoader加载数据,运行images,-labels-=-next(iter(train_loader))报错BrokenPipeError:-[Errno

时间:2022-10-08 11:01:20人气:作者:快盘下载我要评论

事情的经过是这样的;
我正在学习B站up主同济子豪兄的视频Pytorch迁移学习训练自己的图像分类模型【两天搞定AI毕设】

up主使用的Linux系统;我自己使用的本地的windows11系统。
在写到如下代码时;

# 定义数据加载器
from torch.utils.data import DataLoader

BATCH_SIZE = 32

# 分别定义训练集和测试集的数据加载器
# 训练集的数据加载器
train_loader = DataLoader(train_dataset,
                          batch_size=BATCH_SIZE,
                          shuffle=True,
                          num_workers=1
                         )

# 测试集的数据加载器
test_loader = DataLoader(test_dataset,
                         batch_size=BATCH_SIZE,
                         shuffle=False,
                         num_workers=1
                        )

#查看一个batch的图像和标注
# DataLoader 是 python生成器;每次调用返回一个 batch 的数据
images, labels = next(iter(train_loader))

up主那边可以正常运行;我这边报错如下;

win11系统pytorch下使用DataLoader加载数据,运行images,-labels-=-next(iter(train_loader))报错BrokenPipeError:-[Errno

labels

在网上冲浪一段时间之后发现是因为windows系统 pytorch不能使用多进程;num_workers只能设置为0;但是导致GPU训练时速度太慢。

在Windows上;FileMapping对象应必须在所有相关进程都关闭后;才能释放。
启用多线程处理时;子进程将创建FileMapping;然后主进程将打开它。
之后当子进程将尝试释放它的时候;因为父进程还在引用;所以它的引用计数不为零;无法释放。
但是当前代码没有提供在可能的情况下再次关闭它的机会。这个版本官方说num_workers=1是可以用的;更多的线程还在解决;不过现在即便是用2个子进程也已经可以了。

我先将num_workers设为0;报错消失。

还有一种解决方法;可以在不降低线程数量的同时解决这个问题;
在使用DataLoader读取之前加上 if name == ‘main’ : 就可以了。
;这个方法我试了没有成功?这是为什么?;

# 分别定义训练集和测试集的数据加载器
# 训练集的数据加载
if __name__ == ;__main__;:
    train_loader = DataLoader(train_dataset,
                              batch_size=BATCH_SIZE,
                              shuffle=True,
                              num_workers=4
                             )

    # 测试集的数据加载器
    test_loader = DataLoader(test_dataset,
                             batch_size=BATCH_SIZE,
                             shuffle=False,
                             num_workers=4#设为单线程
                            )

    #查看一个batch的图像和标注
    # DataLoader 是 python生成器;每次调用返回一个 batch 的数据
    # images, labels = next(iter(train_loader))
    # print(images.shape)
    # print(labels)

    # DataLoader 是 python生成器;每次调用返回一个 batch 的数据
    # if __name__ == ;__main__;:
    images, labels = next(iter(train_loader))

依然在报同样的错;大家知道的给我留个言~

以上方法参考博客;
DataLoader windows平台下 多线程读数据报错 | BrokenPipeError: [Errno 32] Broken pipe | freeze_support()_Jemary_的博客-程序员信息网

相关文章

  • elasticsearch根据id修改文档的部分数据

    elasticsearch根据id修改文档的部分数据,首先贴上用elasticsearch的语法的案例:然后是直接通过http请求达到修改的目的点...
  • Vue3---Pinia-状态管理(环境搭建安装及各属性使用教程)详细使用教程

    Vue3---Pinia-状态管理(环境搭建安装及各属性使用教程)详细使用教程,vue3 状态管理工具 pinia 使用,Vue3状态管理之Pinia的入门使用教程,Pinia使用详解(vue3默认状态管理),详解Vue3状态管理库Pinia的使用方法,Vue3 使用Pinia状态管理,vue3新一代状态管理器 — pinia的学习与使用,vue3使用pinia管理状态,vue3全家桶之-状态管理器pinia的使用,Vue3中Pinia的基本使用,上手Vue 新的状态管理 Pinia,一篇文章就够了,Pi...

网友评论

快盘下载暂未开通留言功能。

关于我们| 广告联络| 联系我们| 网站帮助| 免责声明| 软件发布

Copyright 2019-2029 【快快下载吧】 版权所有 快快下载吧 | 豫ICP备10006759号公安备案:41010502004165

声明: 快快下载吧上的所有软件和资料来源于互联网,仅供学习和研究使用,请测试后自行销毁,如有侵犯你版权的,请来信指出,本站将立即改正。