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

华欣荣
2023-12-01

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

for epoch in range(start_epoch , opt.nEpochs):
	for iteration, batch in enumerate(training_data_loader, 1):
	   ...
		vgg = Vgg19(requires_grad=False).to('cuda')
		...

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

 类似资料: