0
点赞
收藏
分享

微信扫一扫

PyTorch Dataloader报CUDA error: initialization error的解决方法

柠檬果然酸 2022-04-14 阅读 53
pytorch

这里也就简单记录一下,和这篇博客说的类似:RuntimeError: CUDA error: initialization error_IMU_YY的博客-CSDN博客

除了上面这个报错,还报了:

除了这篇博客的解决方法,还可以直接把num_workers这个参数删掉即可。例如我从

train_dataloader = DataLoader(dataset=train_set, shuffle=True, batch_size=args.batch_size, collate_fn=collate, num_workers=2)

变成了:

train_dataloader = DataLoader(dataset=train_set, shuffle=True, batch_size=args.batch_size, collate_fn=collate)#, num_workers=2

如果性能要求不高,这样似乎也就可以了。

举报

相关推荐

0 条评论