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到如下部分:

pytorch_geometric环境配置二 — pytorch_geometric_第1张图片

点击here,进入如下页面:

pytorch_geometric环境配置二 — pytorch_geometric_第2张图片

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

pytorch_geometric环境配置二 — pytorch_geometric_第3张图片

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

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

pip install torch_geometric

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

你可能感兴趣的:(环境安装,pytorch,深度学习,python)