pytorch在推理时出现CUDA error: out of memory

pytorch在推理时出现CUDA error: out of memory

发现是这个问题

with torch.no_grad():
	output = net(input)