在跑几个月前跑过PyG的GNN模型时,突然当头一棒报错:'NoneType' object has no attribute 'origin',不要慌,赶紧百度 一顿操作猛如虎,发现没几个帖子讲这个,然后发现google后github上的issue有说到,链接为:https://github.com/rusty1s/pytorch_sparse/issues/127
PyG的作者竟然也在上面回答:
意思说可能是版本问题了,印象中木有下载多个torch-sparse呀,后面又不知咋搞的报错说可能和ddl库缺失有关,我寻思可能被我误删什么东西了。。后面看到这作者又说卸载再装试试,我就把pyg卸载了pip uninstall 库名,然后又报错说cuda和pytorch的版本不一致,这个就懂了啊。。。可以用命令nvcc --version查看cuda的版本
想起前面报错的是torch_sparse的ddl库什么缺失的,那就去PyG官网康康PyG安装教程:
对应我的pytorch版本和cuda版本的torch_sparse下载命令:
over,芜湖,起飞!