pyG库安装（cuda）

默认已经安装好了pytorch和cuda环境

1、检查pytorch和cuda版本

 

import torch
print(torch.__version__)   #pytorch版本
print(torch.version.cuda)   #cuda版本

2、以pytorch1.12.*及cuda11.6版本为例下载依赖以及包，其他参照官网

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

3、验证（执行以下不报错则安装成功）

 import torch
 import torch_geometric