PyTorch Geometric安装
PyG是当前最流行和广泛使用的GNN库。
PyG官网
进入官网,点击install,选择如下的方式进行安装:
进入here
安装依赖:
先查看pytorch版本和cuda版本:
import torch
print(torch.__version__)
1.11.0+cu115
print(torch.version.cuda)
11.5
选择对应版本的依赖文件:
进入对应的链接选择正确的python版本和操作系统的依赖:
利用pip进行下载:
pip insatll https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_cluster-1.6.0-cp39-cp39-win_amd64.whl
pip install https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_scatter-2.0.9-cp39-cp39-win_amd64.whl
pip install https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_spline_conv-1.2.1-cp39-cp39-win_amd64.whl
依赖下载完之后,安装PyG
pip install torch-geometric