图神经网络 pytorch_geometric库的安装

前言

它是一个基于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
图神经网络 pytorch_geometric库的安装_第1张图片
我的python是3.6版本,系统是win7
图神经网络 pytorch_geometric库的安装_第2张图片
依次安装下载好的4个依赖包
图神经网络 pytorch_geometric库的安装_第3张图片

二. pytorch_geometric安装

完成第一步之后,再直接使用pip install pytorch_geometric安装pytorch_geometric。

你可能感兴趣的:(python,pytorch,神经网络,深度学习)