关于使用paddlepaddle-gpu训练模型时，发生错误：(Out of memory error on GPU 0. Cannot allocate 49.000000MB ......）_out of memory error on gpu 0. cannot allocate 28.1

作者：你好赵伟 | 2024-04-06 10:22:38

踩

out of memory error on gpu 0. cannot allocate 28.125244mb memory on gpu 0, 1

错误提示内容如下：提示显卡内存不足。

解决方法：

显存不足报错，尝试减少下batch_size、或者裁剪图片等降低下显存占用。或者你不要在本地跑，用高配的服务器去训练后导出模型参数再考虑本地做预测及部署。

我尝试减少batch_size=8（原：64）后，成功运行！！！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/371353