关于yolov5训练时报错cuda内存不足问题

RuntimeError: CUDA out of memory. Tried to allocate 4.08 GiB (GPU 0; 10.76 GiB total capacity; 414.97 MiB already allocated; 4.08 GiB free; 4.85 GiB reserved in total by PyTorch)

当时报这种错误,当然调小batch-size可以解决,但是当不想调小的时候可以把工作线程改小,就能正常训练了关于yolov5训练时报错cuda内存不足问题_第1张图片

在train中有如上图所示的超参数,之前我的工作线程是8,在同等条件下改为4就正常能训练了

你可能感兴趣的:(yolo系列,深度学习,人工智能,python)