训练yolov5时，RuntimeError: CUDA out of memory._yolov5 cuda out of memory

训练yolov5时， 遇到错误。RuntimeError: CUDA out of memory. Tried to allocate 56.00 MiB (GPU 0; 10.00 GiB total capacity; 9.13 GiB already allocated; 0 bytes free; 9.26 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_
split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF。

解决方法一：该博文描述的方法
解决方法二：parser.add_argument(‘–img-size’, nargs=‘+’, type=int, default=[640, 640], help=‘[train, val] image sizes’)
将这里改成parser.add_argument(‘–img-size’, nargs=‘+’, type=int, default=[384, 384], help=‘[train, val] image sizes’)

本来是想将图片尺寸增大一些导致的错误。原因是，我的batch-size仍然是原来的70， 但是图片尺寸增大后， 导致显卡内存不够，还可以将batch-size改小， 我改为32 也是可以跑的。