24.8.22小记 torch.cuda.is_available()返回false的解决办法
出现这个问题后逐一排查。
首先是cuda的安装,看了一下自己电脑上没有安装cuda,于是安装了11.2版本。
安装后还是不对,用
print(torch.version)
print(torch.version.cuda) 也没有确认出错误,于是重新创建了环境,再次安装,再次使用print(torch.version) , print(torch.version.cuda) 发现名称后带有cpu,
发觉安装的版本不对,安装了cpu版本,于是去https://mirrors.aliyun.com/pytorch-wheels/cu111/中手动下载了包,放在虚拟环境项目目录中,直接pip下载,安装上了这些包后返回true。