前言
它是一个基于Pytorch构建的库,可以帮助用户快速构建和训练自己的图神经网络模型。
PyG中实现了很多先进(state of the art)的GNN模块,例如GCN、GraphSage、GAT、SGC、GIN等等。此外,PyG中还包含了大量的benchmark图数据集、丰富的图数据操作方法、支持多GPU。
但是这个库的安装,不能直接使用pip来安装,因为很多依赖库并不会直接安装。很多帖子写的方式也不能安装成功,如下:
一.相关依赖包的安装
需要进入https://data.pyg.org/whl/,根据你的torch版本选择对应版本的依赖包,然后pip安装依赖包whl.
我的环境安装的是torch1.10,没有GPU
我的python是3.6版本,系统是win7
依次安装下载好的4个依赖包
二. pytorch_geometric安装
完成第一步之后,再直接使用pip install pytorch_geometric
安装pytorch_geometric。