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