Pytorch设置多线程进行dataloader

使用 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

你可能感兴趣的:(pytorch)