当前位置:   article > 正文

pytorch模型提示超出内存RuntimeError: CUDA out of memory._htts://wisdomai.xyz/tool/pytorch/archives/2301

htts://wisdomai.xyz/tool/pytorch/archives/2301

跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容,原因是:GPU显存内存不够

简单总结一下解决方法:

  1. 将batch_size改小。
  2. 取torch变量标量值时使用item()属性。
  3. 可以在测试阶段添加如下代码:
		with torch.no_grad():  # 停止自动反向计算梯度
  • 1

参考:
https://wisdomai.xyz/tool/pytorch/archives/2301
https://ptorch.com/news/160.html

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/220577?site
推荐阅读
相关标签
  

闽ICP备14008679号