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 人点赞