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

复制代码
相关推荐
千金裘换酒32 分钟前
LeetCode 移动零元素 快慢指针
算法·leetcode·职场和发展
28岁青春痘老男孩35 分钟前
JDK8+SpringBoot2.x 升级 JDK 17 + Spring Boot 3.x
java·spring boot
北辰alk39 分钟前
RAG索引流程详解:如何高效解析文档构建知识库
人工智能
方璧40 分钟前
限流的算法
java·开发语言
九河云42 分钟前
海上风电“AI偏航对风”:把发电量提升2.1%,单台年增30万度
大数据·人工智能·数字化转型
元Y亨H1 小时前
Nacos - 服务注册
java·微服务
wm10431 小时前
机器学习第二讲 KNN算法
人工智能·算法·机器学习
NAGNIP1 小时前
一文搞懂机器学习线性代数基础知识!
算法
NAGNIP1 小时前
机器学习入门概述一览
算法
曲莫终1 小时前
Java VarHandle全面详解:从入门到精通
java·开发语言