Bug来源:

报错如下:
cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
解决方案:
(1)改小batch_size的值(4或更小),但该方法不太管用
(2)如果经过(1)后又出现如下的报错:
RuntimeError: CUDA out of memory. Tried to allocate 3.03 GiB (GPU 0; 8.00 GiB total capacity; 409.61 MiB already allocated; 5.88 GiB free; 654.00 MiB reserved in total by PyTorch

文章讲述了遇到cuDNN错误CUDNN_STATUS_INTERNAL_ERROR时,尝试减小batch_size的解决方案,以及当出现CUDA内存不足的问题,可能是由于DataLoader的workers设置过多导致,建议减少workers数量以避免内存冲突。
最低0.47元/天 解锁文章
5600

被折叠的 条评论
为什么被折叠?



