YOLOv5训练结束后出现错误代码，查了半天资料，仍然无法解决问题，很难受
cuda error，那么大的提示，你的cuda没有安装好，显卡驱动，cuda和cudnn全部检查一遍看下和pytorch的版本是不是匹配的
