Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB

本文主要是介绍Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB (GPU 0; 6.00 GiB total capacity; 4.44 GiB already allocated; 0 bytes free; 4.49 GiB reserved in total by PyTorch)

可以修改batch_size大小进行解决，例如我是将batch_size = 128改为32即可。

但是这种处理方式也是设备硬伤而折中的办法，因为batch_size缩小有可能影响模型的最终性能。

这篇关于Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---