但我安装的时候明明安装的cuda对应的pytorch
用的清华镜像
命令是conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
然后用测试指令在python下进行测试
torch.cuda.is_available()
返回false说明gpu不能用
网上普遍的问题有三种
(1)你的CUDA,cuDNN,torch版本是否对应
(2)清华镜像下下载的torch
我的属于第二种情况
这种情况不是咱们安装步骤的问题而是conda镜像的问题,
你以为下载的是GPU版本,其实镜像下载的是cpu版本你必须手动去官网用pip下载安装
解决方法
先卸载,pip uninstall torch
我当时用了这个命令,然后又到了anaconda我所安装pytorch的环境目录下又检查还有没有残余
D:\anaconda\envs\(环境名称)\Lib\site-packages
对照着下图如果有一样的删除
正确下载方法
去pytorch官网
因为当时显示我cuda10.2的对应的pytorch windows pip的没有了,我就直接自行找的对应的压缩包下载了
这里面关键的思路就是,你自己手动去官网下载你要找的压缩包,下载,然后你用anaconda prompt 进入环境,然后进入你下载的压缩包的文件路径(cd d:\xx\xx),然后输入 pip install torch==压缩包的全名,就能安装上了
直接进入这个网址然后找你所需对应的东西,要耐心细心
https://download.pytorch.org/whl/torch_stable.html
因为我之前参考了另一篇博客
就下载了这三个
我没装torchaudio,我暂时用不到音频,主要是我不知道载哪个版本的。。。
具体命令与测试请见下图,我成功了
至于下载过程中的时间,你要嫌慢用那个(vpn)
感谢这篇博客给我了很大启发,并且第一个问题在这篇博客上也写了出来
https://blog.csdn.net/qq_46126258/article/details/112708781