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

这里也就简单记录一下，和这篇博客说的类似：RuntimeError: CUDA error: initialization error_IMU_YY的博客-优快云博客

除了上面这个报错，还报了：

During handling of the above exception, another exception occurred:

除了这篇博客的解决方法，还可以直接把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

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