AI环境配置(006)-图神经网络 pytorch_geometric库的安装
pytorch_geometric安装
·
前言
它是一个基于Pytorch构建的库,可以帮助用户快速构建和训练自己的图神经网络模型。
PyG中实现了很多先进(state of the art)的GNN模块,例如GCN、GraphSage、GAT、SGC、GIN等等。此外,PyG中还包含了大量的benchmark图数据集、丰富的图数据操作方法、支持多GPU。
但是这个库的安装,不能直接使用pip来安装,因为很多依赖库并不会直接安装。很多帖子写的方式也不能安装成功,如下:
一.相关依赖包的安装
需要进入https://data.pyg.org/whl/,根据你的torch版本选择对应版本的依赖包,然后pip安装依赖包whl.
我的环境安装的是torch1.10,没有GPU
我的python是3.6版本,系统是win7
依次安装下载好的4个依赖包
二. pytorch_geometric安装
完成第一步之后,再直接使用pip install pytorch_geometric安装pytorch_geometric。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)