CUDA out of memory问题解决_outofmemoryerror: cuda out of memory. tried to all

作者：很楠不爱3 | 2024-02-16 00:41:47

踩

outofmemoryerror: cuda out of memory. tried to allocate 20.00 mib (gpu 0; 4.

在使用resnet demo训练过程中，使用CPU训练不会报错。

但在使用GPU（GTX1650 4GB）训练时提示如下错误：

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 4.00 GiB total capacity; 550.99 MiB already allocated; 1.90 GiB free; 572.00 MiB reserved in total by PyTorch)
1

解决：

改小batch_size
在这里插入图片描述

这里应该时显存不足的原因

将8修改为6就可以了

但修改了以后，[epoch 1]的准确率也下降了：

nw=8时，准确率约91%
nw=6时，准确率约90%
nw=4时，准确率约89%

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/88933?site