问题
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之间的匹配。