我是用python3.8+pytorch1.11,在李沐老师的课程展示中使用的环境是linux没有报错,但win10下在dataloader使用前需要加上
if __name__ == "__main__":
例如:
batch_size = 256
if __name__ == "__main__":
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文章来源:https://www.toymoban.com/news/detail-548402.html
或者return 0 if sys.platform.startswith('win') else 4文章来源地址https://www.toymoban.com/news/detail-548402.html
到了这里,关于D2L,Pytorch win10下使用dataloader时出现runtimeerror: dataloader worker (pid(s) ) exited unexpectedly的解决方案的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!