yolov5 CUDA out of memory

yolov5训练，是1060 batch_size为8

换一台电脑，batch_size为2时，竟然报错：

yolov5 CUDA out of memory

torch版本1.8，后来发现原因是：

workers大于1了，可能开了多进程的缘故。

    parser.add_argument('--workers', type=int, default=1, help='maximum number of dataloader workers')
 

解决方法：把parser中的workers改为1就可以了。

网友的方法，设置subdivisions，我用的yolov5-v5版本，没有这个参数，故没有测试：

解决办法：

1.CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertio `0' failed.

需要修改所使用的模型cfg文件中的subdivision的参数。

由subdivisions=8改成subdivisions=64。

原文链接：https://blog.csdn.net/pts_mjt/article/details/84557681

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/123491678