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

最新推荐文章于 2025-09-16 22:23:36 发布

原创最新推荐文章于 2025-09-16 22:23:36 发布 · 531 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #深度学习

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

查看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

最后直接：