图机器学习 | PyG 安装配置记录

Posted 2022-12-30 叶庭云

PyG（PyTorch Geometric）是构建在 PyTorch 之上的一个库，可以方便地编写和训练图神经网络（GNNs），用于与结构化数据相关的广泛应用。它包括针对图和其他不规则结构的各种深度学习的方法，也称为几何深度学习，来自各种发表的论文。此外，它还包括易于使用的小批量加载器，用于操作许多小的和单个的巨型图，多 GPU 支持，数据管道支持，通过 Quiver 的分布式图学习，大量的通用基准数据集（也可以基于简单的接口创建自己的），GraphGym 实验管理器，以及有用的转换，这些都用于学习任意图以及 3D 网格或点云。

传送门：PyG 官方文档 | Installation

首先需要确认安装了 PyTorch 和配置好 CUDA。

# 确保至少安装了 PyTorch 1.11.0：
python -c "import torch; print(torch.__version__)"
1.11.0+cu113

# 查找安装 PyTorch 的 CUDA 版本:
python -c "import torch; print(torch.version.cuda)"
11.3

python -m pip install --upgrade pip

pip install pyg-lib==0.1.0 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html
pip install torch-cluster==1.6.0 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html
pip install torch-scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html
pip install torch-sparse==0.6.15 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html

# 安装指定版本的 PyG
pip install torch-geometric==2.0.2 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html

---

参考链接：

• pytorch-geometric 与 torch+cu113 匹配的一些依赖库 whl 文件的版本
• PyTorch Geometric (PyG) 安装教程