torch版本对应的torch_geometric与torch-sprse/cluster/scatter库的正确安装

torch_geometric官网:

Installation --- pytorch_geometric documentation

使用上述标红命令即可快速安装需要的包(确定自己环境中安装的pytorch版本以及cuda版本,使用对应的命令即可)

如安装的pytorch为1.60,cuda为10.1,则:

复制代码
pip install torch-scatter torch-sparse torch-cluster -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html

使用以上命令安装相关包即可。

关于核心的torch_geometric包,选择版本之后能看到右边的Quick Star信息,表示其最适合兼容的pytorch版本。但也不是唯一,如我的环境中的相关版本如下,可以正常运行:

复制代码
相关推荐
非优秀程序员5 小时前
第12章 烧钱无底洞!!!不烧就死,烧了也不知道啥时候死|《互联网产品恩仇录》
人工智能
非优秀程序员6 小时前
题记|《互联网产品恩仇录》
后端·程序员·创业
非优秀程序员6 小时前
第05章 播放器大战,暴风、快播与那一夜|《互联网产品恩仇录》
算法
非优秀程序员6 小时前
第04章 邮箱大战,163 与 QQ 的暗战|《互联网产品恩仇录》
java·vue.js·架构
知已难寻6 小时前
Docker 入门:容器技术与 Docker 基础
java·运维·服务器·docker·云原生·容器
Zelman9456 小时前
第06章-交互与观察动作空间
人工智能
冬奇Lab6 小时前
开源项目第258期:Harness² —— Google Research 的「递归式 Agent Harness 优化」,模型权重不动,只改 Agent 的外壳
人工智能·开源
字节跳动视频云技术团队8 小时前
每两家头部手机厂商就有一家选它:火山引擎如何成为 AI 手机助手的“隐形底座”?
人工智能
泯泷8 小时前
像项目经理一样指挥 AI 团队:管理者模式
人工智能·llm·agent