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  比较稳定 其他配置容易出现不识别或者训练无效等问题

你可能感兴趣的:(pytorch,深度学习,神经网络)