torch-geometric(PYG) 环境配置

一、引言

PyTorch Geometric，简称Pyg。Pyg库包含易用的小批量加载器（mini-batch loader）、多GPU支持、大量常见基准数据集和有用的变换，适用于任意图像、三维网格（3D mesh）和点云。基于PyTorch构建，用于处理不规则结构化输入数据（如图、点云、流形）。除了一般的图形数据结构和处理方法外，它还包含从关系学习到3D数据处理等领域中最新发布的多种方法。通过利用稀疏 GPU 加速、提供专用的 CUDA 内核以及为不同大小的输入样本引入高效的小批量处理，PyTorch Geometric 实现了很高的数据吞吐量。都是废话!!! 要不是论文复现用你用的多我早下载隔壁dgl了。

二、环境的配置

这几天快给我逼疯了，作者本地是macbook，还有一个服务器环境(Linux)，安装卸载反复好几次，终于在摸索中找到了一个比较顺利的安装方式。

首先输入python 查看自己的python版本：

 比如作者服务器版本是python3.9.12

下一步是下载pytorch，如果是用服务器GPU，一定能够看好CUDA版本，nvidia-smi就可以看，作者是CUDA11.7, 其实下11.6pytorch也能用。

Start Locally | PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/

 输入上图最下边的指令，就可以下载。

最后一步打开救命网站。

https://github.com/pyg-team/pytorch_geometrichttps://github.com/pyg-team/pytorch_geometric

 拉到这里，打开。

 

 pip list可以查看torch版本，结合CUDA版本，点进去。比如作者是torch1.12.1+cu116。

 cp是python版本，然后选择是windows还是linux。

pip install （torch-scatter复制链接）
pip install （torch-sparse复制链接）
pip install torch-geometric

复制链接如上图所示。 

另外，我的macbook下载的都是cpu版本，原理和上边一样，看好pytorch版本、python版本等等。

最重要是事情，如果安装完导入报错，而且是C++的错误，反方向全卸载重装就行了，无解。