安装该包需要保证pytorch的cuda和torch sparse等包版本对应,推荐离线安装,可以方便对应版本:
- 首先确定本机预装的pytorch版本、cuda版本:
- 在用conda装pytorch的时候是会指定版本的,如果忘记了可以查一下:
- 然后在官网找到对应版本:https://pytorch-geometric.com/whl/
- 点击进去之后下载对应python版本和平台的whl文件,包括torch_cluster、torch_scatter、torch_sparse、torch_spline_conv
使用pip install xxx.whl
离线安装下载好的whl文件 - 然后
pip install torch_geometric
安装torch_geometric
这个时候再去import对应模块就可以正常使用了