PyTorch Geometric Library (简称 PyG) 是一个基于 PyTorch 的图神经网络库,地址为:https://github.com/rusty1s/pytorch_geometric。它包含了很多 GNN 相关论文中的方法实现和常用数据集,并且提供了简单易用的接口来生成图,因此对于复现论文来说也是相当方便。用法大多数和 PyTorch 很相近,因此熟悉 PyTorch 的同学使用这个库可以很快上手。
基于知识图谱和其他图结构的算法常常依赖于此包的安装,而版本的不对应常常导致无法正常使用,基于此,在这我们分享如何根据环境配置相应的安装包。
pip install --no-index torch_scatter -f https://pytorch-geometric.com/whl/torch-1.x.0+cu1xx.html
pip install --no-index torch_sparse -f https://pytorch-geometric.com/whl/torch-1.x.0+cu1xx.html
pip install --no-index torch_cluster -f https://pytorch-geometric.com/whl/torch-1.x.0+cu1xx.html
pip install --no-index torch_spline_conv -f https://pytorch-geometric.com/whl/torch-1.x.0+cu1xx.html
pip install torch-geometric
其中,torch-1.x.0代表的是torch版本,我所使用的是torch-1.7.0,因此这次的x即为7;cu1xx代表的是cuda版本,我的cuda为cuda-11.2,这里即为cu110即可。
如果需要查找在线包的安装,利用search_python_package该网站即可实现。
如果准备离线安装需要的包,访问python_package网站即可。