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
