当前位置:   article > 正文

解决CUDA out of memory. Tried to allocate 11GiB (GPU 0; 32GiB total capacity; 18GiB already_outofmemoryerror: cuda out of memory. tried to all

outofmemoryerror: cuda out of memory. tried to allocate 1.29 gib (gpu 0; 11.

凯哥英语视频
今天用浪潮服务器跑模型遇到一个报错

CUDA out of memory. Tried to allocate 11GiB (GPU 0; 32GiB total capacity; 18GiB already
说是显存不足

一般来说就是 batch_size 过大
最优解决方案, 直接先把batch_size改为1,然后慢慢放大
这样的话, 1可以迅速判断是否是显存问题报错, 2重拾信心啊啊啊

另外,记录一下查看显存的命令和调用运行的命令吧
在这里插入图片描述

别的也没啥说的

ok,那就这样吧~

欢迎各位大佬留言吐槽,也可以深入交流~

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

闽ICP备14008679号