最近学习GNN,需要安装Torch_geometric库,安装了许久,终于成功了,简单记录一下安装过程和踩的坑,以防后续需要
首先通过Anaconda prompt重新安装一个虚拟环境,或者使用之前的虚拟环境,这里我使用的是之前已有的虚拟环境,版本分别为:
Python == 3.7.1;
Torch == 1.5.0;
之前也尝试新建了个虚拟环境,但是import torch总是出错,所以用了已有的正确的虚拟环境进行安装torch-geometric库。
然后在官网上分别下载以下四个文件(CPU,无GPU):

然后按照https://zhuanlan.zhihu.com/p/602615941?utm_id=0 说法,依次使用pip install安装torch_scatter、torch_sparse、torch_cluster和torch_spline_conv(打乱顺序安装可能会出现文件缺失的问题)。
最后 安装:pip install torch-geometric。
然后我按照https://zhuanlan.zhihu.com/p/659534878对profile.py程序进行修改,但会出现其他问题,后来根据https://blog.csdn.net/m0_55684014/article/details/133578645指导,尝试将torch-geometric进行版本降级,即
先卸载已安装的库“pip uninstall torch_geometric”
然后指定版本“pip install torch_geometric==2.0.4”
这时在python环境下运行“import torch” 和“import torch_geometric”均正确,
且通过https://zhuanlan.zhihu.com/p/602615941?utm_id=0里的测试案例,在pycharm里运行得到正确如下输出结果

在pycharm运行时,要记得选择在这个安装了torch_geometric库的python环境下运行

至此,完成了gnn所需的torch_geometric安装!