admin管理员组文章数量:1611505
问题描述:RuntimeError: CUDA out of memory. Tried to allocate 4.69 GiB (GPU 0; 8.00 GiB total capacity; 5.21 GiB already allocated; 997.56 MiB free; 5.27 GiB reserved in total by PyTorch)
问题详情:在一次传入500多样本的时候,显示CUDA内存空间不足,主要在与于数据传入较大,在不能实际提升硬件的条件下可减少一次传入GPU的数据量,即减少batch size
解决方法:减小 batch size 的大小
其他可能的解决方法:
1)减小模型规模
2)测试新样本时使用 torch.no_grad()
等等
(仅供参考,不时补充)
本文标签: allocateMemoryPytorchCUDATotal
版权声明:本文标题:【PyTorch问题】CUDA out of memory. Tried to allocate 4.69 GiB (GPU 0; 8.00 GiB total capacity...略 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/xitong/1728620858a1166341.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论