深度学习CUDA error: out of memory报错记录

运行代码时出现报错

RuntimeError: CUDA error: out of memory

或者

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

或者

untimeError: CUDA out of memory. Tried to allocate 1.55 GiB (GPU 0; 12.00 Gi

出现这个报错的原因是内存不够，可能原因是batchsize太大，读取的数据集太大，开启了多线程。解决方案对应的是减小batchsize，减少读取的数据集，使用单线程。