【图神经网络学习】记录PyG安装问题

努力学习的白衣

于 2025-03-27 18:02:28 发布

阅读量294

点赞数 7

文章标签：学习

本文链接：https://blog.csdn.net/pickle_____/article/details/146566360

版权

本人深度学习小白，有不足欢迎指出，这里仅仅记录一下自己的安装过程

我的cuda是12.1版本

1.安装pytorch

pytorch可以安装cpu版本和gpu版本，我下的gpu版本（吐槽一句gpu版本是真的下得慢如何准确使用国内源高效安装GPU版本的Pytorch - 知乎）

Previous PyTorch Versions | PyTorch

找合适的版本就行，我这里下的2.1.1

2.PyG安装流程

PyG的安装依赖于以下几个核心库：

torch-scatter
torch-sparse
torch-cluster
torch-spline-conv
torch-geometric

pytorch-geometric.com/whl/index.html根据torch和cuda的版去官网找

我这里是

pip install torch-sparse==0.6.18 -f https://pytorch-geometric.com/whl/torch-2.1.1+cu121.html
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-2.1.1+cu121.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-2.1.1+cu121.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-2.1.1+cu121.html
pip install torch-geometric