weixin_42491921的博客

原创 2021-08-27 解决pytorch训练和测试时显存不够的问题

解决pytorch训练和测试事显存不够的问题前言一、使用torch.cuda.empty_cache()删除一些不需要的变量二、使用with torch.no_grad()提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档@TOC前言在使用pytorch训练和测试模型的时候爆显存是很多人都有过的体验，相信新手对这个问题一定很苦恼，原因在于Pytorch 训练时有时候会因为加载的东西过多，这种情况还可以通过cuda的清理技术进行修整，当然如果模型实在太大，那也没办法。废话不多说，下面