CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity; 4.54 GiB already allocated; 14.94 MiB free; 4.64 GiB reserved in total by PyTorch)
在遇到这个问题时
常见的说1.改batch_size的大小2.换好显卡(我电脑都是3070ti还是不行,感觉就不是电脑的事)3.改代码(使用gpu训练的代码就这个就行
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
)
而我在搜这个问题的时候,发现这个老哥的说法很正确,自己也尝试成功了,他的解决办法是