【Pytorch测试时显卡占用率高,GPU利用率低,CUDA out of memory】

Pytorch测试时GPU利用率低,CUDA out of memory


model很小,batchsize=1,

watch -n -1 nvidia-smi;

监视当前GPU使用情况,发现测试时GPU占用爆表,但是利用率只有1%,在测试时加入:

def test:
	model.eval()
	with torch.no_grad():
		test code...
	model.train()

解决,可以正常运行了。

你可能感兴趣的:(pytorch编程,pytorch,gpu,服务器)