需要使用torch_geometric,安装pytorch-sparse失败的解决方案
问题起因是自己需要使用torch_geometric,但是安装完显示需要torch-sparse,
但是装了几次torch-sparse之后都显示失败,经过查询资料知道是版本对应的问题,
参考这篇文章安装对应的依赖包最终搞定:https://www.jianshu.com/p/bef9ab27c415。
安装环境:pycharm 、python3.7
安装步骤
- 安装pytorch:pip install torch;
- 安装torch_geometric:pip install torch_geometric;
- 安装对应的三个依赖包:安装pytorch_geometric包需要安装以下三个包:
torch_sparse、torch_cluster、torch_scatter
这三个包的下载链接如下:
https://pytorch-geometric.com/whl/
我自己的pytorch版本是1.13.0,pytorch-geometric是直接pip 安装的,默认安装的是2.1.0版本, 因为笔记本无gpu,所以点开对应的链接torch-1.13.0+cpu,找到你的环境所支持的依赖包版本,对应我自己的版本,依次下载了
torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
torch_sparse-0.6.15-cp37-cp37m-win_amd64.whl
这三个文件到本地。
分别将以上三个包进行离线安装即可:
pip install E:\torch\torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
pip install E:\torch\torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
pip install E:\torch\torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
全部安装之后就可以正常使用了哦。
注:以上是我自己本地的路径,根据自己的下载路径进行修改即可。
希望对有类似问题的朋友能够有所帮助~