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版本。但也不是唯一,如我的环境中的相关版本如下,可以正常运行:

复制代码
相关推荐
Gigavision32 分钟前
基于BUAA-MIHR数据集的噪声解耦对比学习算法
人工智能·python·深度学习·算法
鹿角片ljp42 分钟前
LeetCode 64:最小路径和复盘|二维 DP 与 ACM 模式完整写法
java·数据结构·算法
IPdodo_1 小时前
跨境 API 调用不稳定怎么办:出口、超时重试与链路监控的实践
网络·python·网络协议
红海云1 小时前
Kimi 双端接入 CloudBase 的工程价值
人工智能·语言模型
Madison-No71 小时前
搭建项目测试环境
linux·运维·服务器·python
计算机编程-吉哥2 小时前
YOLO26 vs YOLO11 vs YOLOv8:深度学习咖啡果实成熟度分割系统【计算机毕业设计选题推荐】
人工智能·python·深度学习·yolo·django·毕业设计
冬奇Lab2 小时前
一年前没启动 AI 提效的团队,今年在付什么钱?
人工智能
彧azz2 小时前
算法设计与分析:贪心与动态规划
数据结构·学习·算法·贪心算法·动态规划
泡海椒2 小时前
PDF 表格样式优化:jquick-pdf 边框、圆角、背景色
java·开发语言·pdf