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

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

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

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