图神经网络 torch_geometric 依赖包的安装

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网站即可。

  • 10
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值