PyTorch Geometric (PyG) 库的安装_pyg安装

前言

最近在学习图神经网络，需要用到PyG库，发现这个库的安装不能简单的使用pip install 安装，这里记录一下。

1、PyG库的安装过程

第一步：查看自己的torch版本和cuda版本

import torch
print(torch.__version__)
print(torch.version.cuda)
>>2.0.0+cu117
1
2
3
4

以本人的pytorch为例，pytorch版本是2.0.0，cuda版本是11.7。

第二步：安装相关依赖
点击进入下面网站，https://pytorch-geometric.com/whl/。
点开网站，找到自己的torch和cuda对应的依赖，本人的torch版本是2.0.0，cuda版本是cu117，如下所示：

找到对应版本并点击进入，显示如下，复制网址备用。

接着打开Anaconda Prompt，conda activate pytorch进入到自己的pytorch环境中，在pytorch环境中执行以下命令：

pip install torch_scatter -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
1

pip install torch_sparse -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
1

pip install torch_cluster -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
1

pip install torch_spline_conv -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
1

注意，-f后面跟的是之前我们复制的网址，即上一个图中画着红线的网址，这个网址是和自己的pytorch版本和cuda版本对应的，每个人的都不一样。
最后执行：

pip install torch-geometric
1

至此，安装过程结束。

2、版本问题解决

如果之前版本安装错了，可以卸载重新安装，卸载命令是：

pip uninstall torch_geometric torch_scatter torch_sparse torch_cluster torch_spline_conv
1