cuda和pytorch不兼容,多半是版本不匹配,可以通过官网查得下载的cuda版本的pytorch。
我的cuda版本为11.4,通过网上查的匹配的pytorch版本应为1.8or1.9.
操作:
torch.tensor(data,device='cuda')报错
然后查看pytorch与cuda是否兼容
print(torch.cuda.is_available())
输出False
查看pytorch的版本
print(torch.__version__)
发现为1.6,版本过低
进入anaconda prompt
更新pytorch
activate pytoch #激活我创建的虚拟环境pytorch
conda update pytorch torchvision
更新完后发现更新完的版本为1.72(使用清华镜像网站可能版本没有最新),但也能与cuda11.0兼容
输入print(torch.cuda,is_available())
返回Ture
解决!