admin管理员组

文章数量:1611548

问题描述：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://www.elefans.com/xitong/1728620858a1166341.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。