pytorch_geometric环境配置二 — pytorch_geometric

根据pytorch_geometric环境配置一安装好pytorch之后，便可安装pytorch_geometric进行GNN的学习了。

接下来介绍pytorch_geometric如何进行安装。

传送门：GitHub - pyg-team/pytorch_geometric: Graph Neural Network Library for PyTorch

进入后，ctrl+f  wheel到如下部分：

点击here，进入如下页面：

 根据自己安装的torch版本和cuda版本（如无安装cuda则选择cpu）进入链接，进入后，共需下载4个wheel文件：

以cluster为例，首先是版本选择，随后是python环境，如cp36表示Python3.6，随后是操作系统，此处主要根据自己的python版本和操作系统进行选择即可，四个包下载到本地后，通过pip进行安装。

以上依赖安装完成后，运行如下代码即可安装成功。

pip install torch_geometric

 至此，pytorch_geometric环境搭建成功，大家可以开始GNN的学习了！