Pytorch训练时GPU利用率为0或很低

2022-09-24 08:55:18

pytorch训练时，显存明显上去了，模型和数据都在gpu上，但GPU利用率仍为0%
解决方法：
增加dataloader中的num_workers

  train_loader= DataLoader(dataset=train_dataset,
                                 batch_size=512,
                                 shuffle=False, num_workers=6)
  • 作者：shimmer_ghq
  • 原文链接：https://blog.csdn.net/m0_47702386/article/details/125710297
    更新时间：2022-09-24 08:55:18