使用 GPU 进行训练的时候,
torch.utils.data.DataLoader
中的 num_workers
设置为0,设pytorch 自已决定
参考文献
(1条消息)Pytorch设置多线程进行dataloader时影响GPU运行_xiuxiuxiuxiul的博客-CSDN博客
https://blog.csdn.net/xiuxiuxiuxiul/article/details/86233500
(1条消息)PyTorch——num_workers参数的设置_墨门-CSDN博客
https://blog.csdn.net/songyuc/article/details/103365591