pytorch_geometric安装教程

PyG 安装
PyG 全称是PyTorch-Geometric，是一个PyTorch基础上的一个库，专门用于图形式的数据，可以加速图学习算法的计算过程，比如稀疏化的图等。

先查看版本：

import torch;

print(torch.__version__)

print(torch.version.cuda)

:

注意 cuda版本必须是10.0以上

对于 PyTorch 版本为1.4.0，安装一下库：

其中${CUDA}取决于为本地的PyTorch安装环境，可以替换为：cpu, cu92, cu101 or cu102

#其中${CUDA}取决于为本地的PyTorch安装环境，可以替换为：cpu, cu92, cu101 or cu102
pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-cluster==1.4.5
pip install torch-geometric

或者

$ pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
$ pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
$ pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
$ pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
$ pip install torch-geometric

linux下 安装

进入官网找适合自己的版本：
https://pytorch-geometric.com/whl/torch-1.4.0.html

找到合适的版本下载到本地，然后再pip安装。

torch_cluster-latest+cu101-cp37-cp37m-linux_x86_64.whl

torch_scatter-latest+cu101-cp37-cp37m-linux_x86_64.whl

torch_sparse-latest+cu101-cp37-cp37m-linux_x86_64.whl

torch_spline_conv-latest+cu101-cp37-cp37m-linux_x86_64.whl

pip install torch-geometric