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

最新推荐文章于 2024-08-21 19:53:25 发布

ASS-ASH

最新推荐文章于 2024-08-21 19:53:25 发布

阅读量5.9k

点赞数 2

分类专栏： Python与相关环境文章标签：算法 python 人工智能 java 程序员

本文链接：https://blog.csdn.net/qq_38563206/article/details/133931835

版权

41 篇文章 1 订阅

订阅专栏

本文指导如何快速安装适合特定PyTorch（如1.6.0）和CUDA（如10.1）版本的torch_geometric包及其依赖。安装过程中提到的QuickStart信息有助于确保兼容性，即使在不同版本组合下也能正常运行。

摘要由CSDN通过智能技术生成

torch_geometric官网：

使用上述标红命令即可快速安装需要的包（确定自己环境中安装的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版本。但也不是唯一，如我的环境中的相关版本如下，可以正常运行：

关注

专栏目录