CUDA out of memory问题解决

在使用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)

解决:

改小batch_size
CUDA out of memory问题解决_第1张图片

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

将8修改为6就可以了

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

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

你可能感兴趣的:(#,PyTorch,深度学习,pytorch,神经网络)