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 比较稳定 其他配置容易出现不识别或者训练无效等问题
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!
