CUDA out of memory

在训练深度学习模型时，我遇到了这个bug
CUDA out of memory

这个bug意思就是显存不足，有两种办法可以解决。

1、杀死线程
具体操作可参见
杀死GPU线程

2、减小batch-size
这个比较容易修改
batch-size即每次读进显存的数据（批）数量
例如，我之前的batch-size为12，导致显存不足，修改为8，顺利运行。
当然，这样修改也会导致训练速度变慢，因此具备一块好显卡是很有必要的。

文章来源: zstar.blog.csdn.net，作者：zstar-_，版权归原作者所有，如需转载，请联系作者。

原文链接：zstar.blog.csdn.net/article/details/117392649