CUDA out of memory 2022-06-14 10:50:21 浏览数 (1) 在训练深度学习模型时,我遇到了这个bug CUDA out of memory这个bug意思就是显存不足,有两种办法可以解决。1、杀死线程 具体操作可参见 杀死GPU线程2、减小batch-size 这个比较容易修改 batch-size即每次读进显存的数据(批)数量 例如,我之前的batch-size为12,导致显存不足,修改为8,顺利运行。 当然,这样修改也会导致训练速度变慢,因此具备一块好显卡是很有必要的。 批量计算 编程算法 0 人点赞 上一篇:分享雷军22年前编写的代码