解决Pytorch 训练与测试时爆显存(out of memory)的问题

1.该资源内容由用户上传，如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款（资源遇到问题，请及时私信上传者）

版权申诉

5星 · 超过95%的资源 4 下载量 173 浏览量 2021-01-01 15:06:47 上传评论 1 收藏 35KB PDF 举报

温馨提示

试读

1页

Pytorch 训练时有时候会因为加载的东西过多而爆显存，有些时候这种情况还可以使用cuda的清理技术进行修整，当然如果模型实在太大，那也没办法。使用torch.cuda.empty_cache()删除一些不需要的变量代码示例如下： try: output = model(input) except RuntimeError as exception: if out of memory in str(exception): print(WARNING: out of memory) if hasattr(torch.cuda, 'empty_cache'):

资源推荐

资源评论