解决Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocatio

直接上图:

最近在训练一个语音识别程序的时候,执行训练脚本直接报错:

Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

经过网上搜索,这种一般是由于GPU显存不够导致的原因,我的解决办法是直接调小了batch_size的大小,将batch_size的值由原来的16修改为了8,即可正常运行

发布了96 篇原创文章 · 获赞 96 · 访问量 22万+

猜你喜欢

转载自blog.csdn.net/qq_33221533/article/details/100188050
今日推荐