PyTorch-Geometric安装教程

最近在做预测方面的研究,然后就需要用到GCN。

PyG (即PyTorch-geometric)是德国多特蒙德工业大学的研究者们提出的一个基于Pytorch的拓展库,为各种复杂的图神经网络封装好了一个统一的规范接口,为我们搭建自己设计的图神经网络提供了便利。

为了安装这个整的是踩了无数的坑,试了多个版本,真的是令人心累,现在简单说一下我的安装过程,避免大家踩坑!首先上个两个链接,后面就需要用到了:pytorch的版本
torch1.4中PyTorch-Geometric的依赖包

一,依赖包的下载

torch-scatter 、torch-sparse 、torch-cluster 、torch-spline-conv 这4个包装了才可以安装PyTorch-Geometric。

我在这里给大家推荐的是离线安装,在线安装的时候很麻烦,会提示多个版本,他不能自己选择合适的版本。
在线安装所以推荐大家把对应版本的依赖包下载后安装。

首先在对应pytorch的版本这个链接里面选择自己的torch版本,我的是1.4,所以选择图中的这个

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值