简单教程！一分钟无风险安装pytorch_geometric

最新推荐文章于 2025-07-11 20:54:30 发布

玉章.

最新推荐文章于 2025-07-11 20:54:30 发布

阅读量464

点赞数 1

CC 4.0 BY-SA版权

文章标签： pytorch python 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_56677458/article/details/131010397

这篇文章指导读者如何查看Torch的CUDA版本和torch的版本，并详细说明了如何根据特定的CUDA和Torch版本找到并安装PyG的依赖包，包括torch-scatter,torch-sparse,torch-cluster和torch-spline-conv。最后，文章演示了如何安装torch-geometric库，并提供检查安装成功的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

查看torch版本

import torch
print(torch.version.cuda)"
print(torch.__version__)

查找对应的pyg依赖包

前往这个链接：https://data.pyg.org/whl/
比如说我的cuda_version是11.3，torch版本是1.12.0，那么找到torch-1.12.0+cu113这个版本，点进去，里面就是pyg所需要的所有的依赖包,复制这个网站的链接。

安装方法

-f后面跟的是你刚刚复制的链接。

pip install --no-index torch-scatter -f https://data.pyg.org/whl/torch-1.12.0%2Bcu113.html
pip install --no-index torch-sparse -f https://data.pyg.org/whl/torch-1.12.0%2Bcu113.html
pip install --no-index torch-cluster -f https://data.pyg.org/whl/torch-1.12.0%2Bcu113.html
pip install --no-index torch-spline-conv -f https://data.pyg.org/whl/torch-1.12.0%2Bcu113.html

最后直接：