CUDA out of memory

Posted Z|Star

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了CUDA out of memory相关的知识,希望对你有一定的参考价值。

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

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

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

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

以上是关于CUDA out of memory的主要内容,如果未能解决你的问题,请参考以下文章

CUDA out of memory windows cmd

尝试验证安装时,首次安装 TensorFlow 会出错。 CUDA_ERROR_OUT_OF_MEMORY

Tensorflow GPU错误CUDA_ERROR_OUT_OF_MEMORY:内存不足

未能从设备分配 158.06M(165740544 字节):CUDA_ERROR_OUT_OF_MEMORY

CUDA out of memory

出现 CUDA out of memory 的解决方法