当前位置:   article > 正文

D2L,Pytorch win10下使用dataloader时出现runtimeerror: dataloader worker (pid(s) ) exited unexpectedly的解决方案_dataloader worker (pid(s) 16632) exited unexpected

dataloader worker (pid(s) 16632) exited unexpectedly

        我是用python3.8+pytorch1.11,在李沐老师的课程展示中使用的环境是linux没有报错,但win10下在dataloader使用前需要加上

if __name__ == "__main__":

例如:

  1. batch_size = 256
  2. if __name__ == "__main__":
  3. train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size)

即可成功运行

        网上还有其他方法将num_worker设为0也可以单线程成功运行,不过直接导入d2l的dataloader没有对应的参数

可能原因:

        程序在运行时启用了多线程,而多线程的使用用到了freeze_support()函数。
freeze_support()函数在linux和类unix系统上可直接运行,在windows系统中需要跟在main后边。

更新:可以直接改包内代码

        找到此文件…\Anaconda\Lib\site-packages\d2l\torch.py
        编辑其中的def get_dataloader_workers():
        改为return 0

        或者return 0 if sys.platform.startswith('win') else 4

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/562702
推荐阅读
相关标签
  

闽ICP备14008679号