python安装pyg（pytorch_geometric）的两种方式

冷不防 2023-10-04 21:12 29阅读 0赞

#### 文章目录 ####

*  PyTorch Geometric简介
 *  查看环境信息
 *  1. 通过conda安装
 *  2. 通过pip安装

## 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__)"

![在这里插入图片描述][32ddd0f6dc0b4a76a28176095fc0bd57.png]

**查看cuda版本**

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

![在这里插入图片描述][af9dbfdfd69b4aa692c2e025e2d3ec3c.png]

## 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

![在这里插入图片描述][1aa46f395f29487f8cc0b8f1a0782a11.png]

安装成功！

[32ddd0f6dc0b4a76a28176095fc0bd57.png]: https://img-blog.csdnimg.cn/32ddd0f6dc0b4a76a28176095fc0bd57.png
[af9dbfdfd69b4aa692c2e025e2d3ec3c.png]: https://img-blog.csdnimg.cn/af9dbfdfd69b4aa692c2e025e2d3ec3c.png
[1aa46f395f29487f8cc0b8f1a0782a11.png]: https://img-blog.csdnimg.cn/1aa46f395f29487f8cc0b8f1a0782a11.png