怎么推理还cuda out of memory

with torch.no_grad():
	y = model(x)

记得加上with torch.no_grad()表示不计算梯度。