Pytorch CUDA out of mermory问题方案

先使用with torch.no_grad():然后使用：torch.cuda.empty_cache()删除部分缓存
具体代码如下：

with torch.no_grad():
	torch.cuda.empty_cache()
    checkpoint = torch.load(args.checkpoint)#载入读取的内容

亲测可以～