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