安装PyTorch Geometric(PyG)教程

安装 & 测试 PyG

初始环境：PyTorch 1.10.0 Python 3.8 Cuda 11.3

安装步骤

Step 0:

安装之前强烈建议检查版本：

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

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

Step 1：

选择自己对应的版本，这里建议和官方文档的版本保持一致。（windows和Linux的版本是一样的同样可以选择图中的配置）

如：Pytorch 1.10.0 CUDA 11.3

注意：不要使用conda命令直接安装，这样容易造成版本无法对齐。

Step 2：

通过pip wheels进行安装

pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

这里将 ${CUDA} 和 ${TORCH} 改为你的环境中的CUDA版本和PyTorch版本

如：pytorch 1.10.0 和 CUDA 11.3

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

Step 3：

测试是否安装成功：

import torch
from torch_geometric.data import Data

edge_index = torch.tensor([[0, 1, 1, 2],
                           [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = Data(x=x, edge_index=edge_index)
>>> Data(edge_index=[2, 4], x=[3, 1])