python3.7+pytorch1.7.1+cuda10.2环境安装torch_geometric

1、官网上查看自己的环境对应的torch_geometric版本

（1）github网址：https://github.com/rusty1s/pytorch_geometric

（2）打开网站之后，能在网页上看到  “Installation”，我用到的版本如下图：

2、安装torch_geometric的命令

我的电脑配置是cuda10.2,对应的是cu102（如果不是这个环境，可以更改下面命令中加粗的内容），在命令行中运行如下代码即可：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

pip install torch-geometric

 3、是否需要其他依赖包？

在运行上述命令的过程中，我发现系统会自动下载一些包，通过查看原项目，可以看到有一个setup.py文件，打开后可以看到下图，numpy,  tqdm等都是本项目的依赖包。如果之前没有下载过这些包也不用担心，直接运行第二步骤的代码，系统会自动安装需要的包。

4、安装出现问题，未安装上torch-cluster和torch-scatter

可以打开网址https://pytorch-geometric.com/whl/torch-1.7.0.html(如果是其他版本的torch，使用该网址查看对应版本https://pytorch-geometric.com/whl/)，找自己需要的版本下载之后放在“script"文件夹下，也可以自建文件夹，我的虚拟环境的配置是python3.7.9，pytorch版本是1.7.1，查找到对应的兼容标签（Compatible tags）是”cp37-cp37m-win_amd64“，因此下载了下图所示的版本。

总结安装步骤：

• 下载合适的torch-cluster版本，并放在“script"文件夹下
• cd到该“script"文件夹
• 输入命令pip install torch_cluster-1.5.8+cpu-cp37-cp37m-win_amd64.whl，并回车
• 显示Successfully......       安装成功!!!

 

注意：查找到对应的兼容标签的方法：参考https://blog.csdn.net/happywlg123/article/details/107281936

　　我们可以通过下面这个经过测试有效的命令，来查看当前Python版本匹配那些文件。
　　有效命令如下（直接在命令行中输入，不要在Python解释器输入哦）：

　　　　　　pip debug --verbose

　　结果很长，但是我们在这里需要的是”Compatible tags"后面的内容，如下，红圈中就是我们需要的版本。

 

 

一定要注意版本的匹配，如果使用cu102版本的torch-geometric，就要使用cu102版本的torch-cluster和torch-scatter