下载了pytorch和cuda但是torch.cuda.is_available()为false

原本找了很久原因，不知道为什么下载了pytorch和cuda但是torch.cuda.is_available()为false，后面在查看torch包的时候发现时cpu版本，大家安装pytorch要按照官网上面的的安装。例如安装cuda11.8：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

而不要直接pip install torch，如果你不确定自己是不是这个原因可以通过

conda list

查看自己的安装包

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-a1KgIJgE-1686626373915)(data/attachment/forum/202306/13/111450mv13z8zgvvz081lg.png?imageMogr2/auto-orient/strip%7CimageView2/2/w/300 “image.png”)]

这种显示才是cuda版本

解决方案就是 先remove掉原本torch然后重新安装cuda版本torch
原文地址