PyTorch Geometric简介

PyTorch Geometric是一个处理图结构数据的计算库，它使用PyTorch作为其后端计算平台，提供了丰富的图算法和深度学习模型，如图卷积网络、图注意力网络等。PyTorch Geometric的设计和实现旨在提高图深度学习的性能和效率，特别是在处理大规模图结构数据时。主要功能包括：

1. 数据集和数据转换：PyTorch Geometric提供内置的数据集和数据转换模块，方便用户自定义数据集和数据预处理方式。

2. 图卷积和图注意力网络：PyTorch Geometric支持使用图卷积网络（GCN）和图注意力网络（GAT）等常见的图神经网络模型，以便用户能够轻松地构建自己的深度学习模型。

3. 大规模图处理：PyTorch Geometric使用高效的图数据存储格式，支持GPU加速，并提供了大量的图剪枝和分区等算法，从而提高处理大规模图模型的性能和效率。

4. 可扩展性和灵活性：PyTorch Geometric基于PyTorch，可以与其它常用的深度学习库无缝集成，同时具有很好的可扩展性和灵活性，可以自定义算法和模型实现方式。

总之，PyTorch Geometric是一个功能丰富、高性能、高灵活性的图机器学习计算库，特别适合处理大规模、复杂的图结构数据。

查看环境信息

查看pytorch版本

python -c "import torch; print(torch.__version__)"

查看cuda版本

python -c "import torch; print(torch.version.cuda)"

1. 通过conda安装

conda install pyg -c pyg

报错的话试试下面的：

conda install pyg -c pyg -c conda-forge

2. 通过pip安装

pip install pyg-lib torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.10.2+cu113.html

pip install torch-geometric

pip install torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-1.10.2+cu113.html

安装成功！