pytorch显存不够用的时候怎么处理

首先把不需要求导数的模型放入with_no_grad中

不用BN层和dropout的用eval()

delforward的中间变量损失函数，图像和标签。