torch-geometric一直装不上,因为四个依赖包一直装不上(torch_cluster、torch_scatter、torch_sparse、torch_spline_conv),网上看了好多教程,一直都是各种fail bulid wheel
最后还是自己手动下载whl文件,然后使用pip install + whl路径成功安装,下面是方法:
0.如果之前失败安装了4个依赖包,那么先卸载掉
pip uninstall torch-scatter torch-sparse torch-cluster torch-spline-conv
1.查看自己的pytorch和cuda版本,例如torch-1.10.0+cu113
2.打开pyg官方依赖下载页data.pyg.org/whl/
3.找到自己版本的torch和cuda,例如torch-1.10.0+cu113
4.点开以后,分别下载四个依赖包,这里选择自己的python版本以及操作系统,我这里以python3.7+windows为例。
注:这里可能会有多个版本,比如cluster就有1.6.0和1.5.9,我是选了更新一点的那个,应该两个版本的都是可以用的
5.安装依赖包,使用pip install + whl文件,分别安装四个依赖包
pip install C:\Users\Administrator\Desktop\4whl\torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
pip install C:\Users\Administrator\Desktop\4whl\torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
pip install C:\Users\Administrator\Desktop\4whl\torch_sparse-0.6.13-cp37-cp37m-win_amd64.whl
pip install C:\Users\Administrator\Desktop\4whl\torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl
6.最后,安装pyg
pip install torch-geometric==x.x.x
7.安装完毕,我们import测试一下:
python
import torch_geometric
exit()