GPU利用率(gpu-util)低下的可能原因

2022-09-27 09:59:17

本文章分享一下自身遇到的一个问题,参考很多其他文章对GPU利用率低的解答发现自己都解决了,可是利用率还是一会儿10%一会儿90%。我的原因是使用了VGG模块用于计算感知损失。

for epochinrange(start_epoch, opt.nEpochs):for iteration, batchinenumerate(training_data_loader,1):...
		vgg= Vgg19(requires_grad=False).to('cuda')...

正确的做法是把感知损失的加载放在epoch外面,加载一次即可。

  • 作者:生成滞涨网络~
  • 原文链接:https://blog.csdn.net/qq_42061298/article/details/122373487
    更新时间:2022-09-27 09:59:17