pytorch模型提示超出内存模型提示超出内存RuntimeError: CUDA out of
memory.
跑模型时出现跑模型时出现RuntimeError: CUDA out of memory.错误错误 查阅了许多相关内容，原因是：GPU显存内存不够显存内存不够
简单总结一下解决方法：简单总结一下解决方法：
将batch_size改小。
取torch变量标量值时使用item()属性。
可以在测试阶段添加如下代码：
with torch.no_grad(): # 停止自动反向计算梯度
参考：参考：
https://wisdomai.xyz/tool/pytorch/archives/2301
https://ptorch.com/news/160.html
作者：菜叶儿掉啦
评论0