RuntimeError: CUDA out of memory. Tried to allocate 100.00 MiB (GPU 0; 5.78 GiB total capacity; 3.92

RuntimeError: CUDA out of memory. Tried to allocate 100.00 MiB (GPU 0; 5.78 GiB total capacity; 3.92 GiB already allocated; 89.62 MiB free; 3.96 GiB reserved in total by PyTorch)

训练yolov5的时候出现该问题
解决方法:
把bath调小即可,因为溢出gpu的计算能力

你可能感兴趣的:(机器视觉,ubuntu,深度学习,pytorch,python)