一、前言

我们使用torch_geometric库来实现图神经网络的编码，因为它与PyTroch天然集成。本文介绍了PyTorch Geometric（PyG）的安装与测试。

二、安装

首先打开官方的安装说明文档：https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

如下图所示，需要根据PyTroch与CUDA（如有）版本来修改红线部分对应的安装指令：

打印PyTorch版本：

import torch

print(torch.__version__)
print(torch.version.cuda)

结果：

1.12.1+cpu
None

作者的安装指令如下：

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.12.1+cpu.html

三、测试

安装完成后，我们引入KarateClub数据集，看是否能正常使用：

from torch_geometric.datasets import KarateClub

dataset=KarateClub()
print(dataset.num_features)
print(dataset.num_classes)
print(dataset.data)

结果：

34
4
Data(x=[34, 34], edge_index=[2, 156], y=[34], train_mask=[34])

测试成功~