RuntimeError: CUDA error: out of memory CUDA kernel errors might be asynchronously reported at some
fuser -v /dev/nvidia*查看进程,使用kill -9 <进程号>或者直接把所有运行的程序关闭,然后再重新打开,就可以进行训练了。
·
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 <进程号>
或者直接把所有运行的程序关闭,然后再重新打开,就可以进行训练了
更多推荐

所有评论(0)