yolov5训练(train)的时候 P R 值为0

问题:

在版本是py3.9+torch1.9+cuda11.1 的环境下  训练模型的时候P R 值一直是0  

解决方案:

卸载pytorch  重新去官网装  CUDA版本改成10.2    即可 

以conda 为例  torch1.9+cuda10.2 安装代码如下

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

本人目前用的比较稳定的环境是:

py3.9+torch1.9+cuda10.2  比较稳定 其他配置容易出现不识别或者训练无效等问题


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部