PyTorch-Geometric安装

PyTorch-Geometric安装

本文借鉴了知乎的回答

---

问题：
在如下环境中：

1. 操作系统：Win10
2. Cuda版本：Cuda 10.2
3. Pytorch版本：Pytorch 1.6
4. Python版本：Python 3.7

已经安装好pytorch的GPU版本，要安装pytorch-geometric。

---

步骤：
若想安装pytorch-geometric，需要安装以下包：

1. torch-scatter
2. torch-sparse
3. torch-cluster
4. torch-spline-conv
5. torch-geometric

第5个包可以直接用pip安装。
前4个包直接用pip安装会报错。

---

解决方法：
直接用whl文件进行安装。
该项目提供了各个版本的whl文件。
并且介绍了各个版本的whl文件使用方法。
如果要总结一下作者在项目中介绍的方法，简而言之就是如下命令：

pip install ${PackageName} == latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html

其中${PackageName} 是前四个包的名称，即torch-scatter，torch-sparse，torch-cluster，torch-spline-conv；${CUDA} 是CUDA的版本号，可以从cpu，cu92，cu101，cu102中选择。