qq_52628753的博客

原创 关于cuda和pytorch的一些问题

首先我根据我的显卡驱动，我安装cuda官网给我推荐的11.5版本的cuda，接着我去安装pytorch 没找到对应11.5的版本，最高只找到11.3的，我又降级成11.3，然后又发现最高对应版本是python3.9的，我是3.10版本，我又重装python，最后对训练集进行训练发现训练过程中P，R显示为Null，模型也预测不了，然后去网上搜，发现是可能是pytorch版本不合适，我又降级pytorch，cuda也降成10.2，终于能跑了。还有我用conda下载cuda对应的pytorch，下出来是cpu版本