PyG 库的安装
- 前言
- 1、PyG库的安装过程
- 2、版本问题解决
前言
最近在学习图神经网络，需要用到PyG库，发现这个库的安装不能简单的使用pip install 安装，这里记录一下。
1、PyG库的安装过程
第一步：查看自己的torch版本和cuda版本
import torch
print(torch.__version__)
>>2.0.0+cu117
以本人的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 torch_sparse torch_cluster torch_spline_conv -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
注意，-f后面跟的是之前我们复制的网址，即上一个图中画着红线的网址，这个网址是和自己的pytorch版本和cuda版本对应的，每个人的都不一样。
最后执行：
pip install torch-geometric
至此，安装过程结束。
2、版本问题解决
如果之前版本安装错了，可以卸载重新安装，卸载命令是：
pip uninstall torch_geometric torch_scatter torch_sparse torch_cluster torch_spline_conv
