3090显卡 torch.cuda.is_available()返回false的解决办法

问题

1.执行Nvidia-smi 命令没有报错，能够显示驱动信息；
2.执行 torch.backends.cudnn.enabled is TRUE
3.torch.cuda.is_available()一直返回False

解决

把torch，torchvision等相关安装包全部删除，安装适合版本的torch。

30系列显卡是新一代架构，新驱动不支持cuda9以及cuda10，所以必须安装cuda11、而pytorch现在稳定版为1.6，最高仅支持到cud10.2。所以唯一的办法就是使用上处于beta测试的1.7或1.8。这也是为啥一开始就强调本文的写作时间，因为未来一定会有稳定版发布，请使用稳定版。torch_geometric 计算包官方发布的现成轮子仅支持到pytorch-1.6和cuda10.2。
所以核心问题就是cuda11 pytorch 以及torch_geometric之间的匹配。