RuntimeError: CUDA error: out of memory CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.

训练yolov5时报的错,主要是因为GPU的显存占用满了,可以使用命令gpustat查看使用情况,然后在fuser -v /dev/nvidia*查看进程,使用kill -9 <进程号>

或者直接把所有运行的程序关闭,然后再重新打开,就可以进行训练了

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐