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:内存不足