yolov5显存溢出

Question

运行train.py的时候总是现存溢出，然后修改了batch_size也无济于事，想问一下大家这个问题是怎么解决的呀？

Answer 1

解决方案：

训练yolov5时出现RuntimeError: CUDA out of memory_XINFINFZ的博客-CSDN博客 pytorch明明有多余显存却不分配，cuda溢出的正确解决方法（错误方法：调小batch_size） https://blog.csdn.net/weixin_43945848/article/details/126266421