前言

它是一个基于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。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐