Pytorch 训练与测试时爆显存(out of memory)的一个解决方案

Pytorch 训练时有时候会由于加载的东西过多而爆显存，有些时候这种状况还能够使用cuda的清理技术进行修整，固然若是模型实在太大，那也没办法。python 使用torch.cuda.empty_cache()删除一些不须要的变量代码示例以下：测试 try: output = model(input) except RuntimeError as exception: if "o

>>阅读原文<<