需要使用torch_geometric,安装pytorch-sparse失败的解决方案

这里写自定义目录标题

  • 需要使用torch_geometric,安装pytorch-sparse失败的解决方案
    • 安装环境:pycharm 、python3.7
    • 安装步骤

需要使用torch_geometric,安装pytorch-sparse失败的解决方案

问题起因是自己需要使用torch_geometric,但是安装完显示需要torch-sparse,
但是装了几次torch-sparse之后都显示失败,经过查询资料知道是版本对应的问题,
参考这篇文章安装对应的依赖包最终搞定:https://www.jianshu.com/p/bef9ab27c415。

安装环境:pycharm 、python3.7

安装步骤

  1. 安装pytorch:pip install torch;
  2. 安装torch_geometric:pip install torch_geometric;
  3. 安装对应的三个依赖包:安装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

全部安装之后就可以正常使用了哦。
注:以上是我自己本地的路径,根据自己的下载路径进行修改即可。
希望对有类似问题的朋友能够有所帮助~

你可能感兴趣的:(笔记,pytorch,python,深度学习)