admin管理员组

文章数量:1611561

 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')

）

而我在搜这个问题的时候，发现这个老哥的说法很正确，自己也尝试成功了，他的解决办法是

关于gpu使用不了bug的修复_torch无法使用gpu_雾淞沆砀的博客-CSDN博客

本文标签： MIBallocateCUDAMemoryGPU

版权声明：本文标题：CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0； 6.00 GiB total capacity； 4.54 GiB already a 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://www.elefans.com/xitong/1728620076a1166239.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。