图卷积中常用的一个函数,直接pip install下载不下来,查阅各种资料之后跑通的一个比较方便的方法,记录一下:
1. 查看torch以及cuda版本:
import torch
print(torch.__version__)
print(torch.version.cuda)
2. 找版本都对应上的torch_scatter并下载:
https://pytorch-geometric.com/whl/
cp是指python版本,可以在命令行中输入python查看。
例如我python版本为3.8,下载torch_scatter-2.0.9-cp38-cp38-win_amd64.whl
3. 安装:
在命令行中:
pip install 下载的whl文件所在路径
完成!
如果安装完成后import torch_scatter报错:
OSError: [WinError 127] 找不到指定的程序。
说明还是下载的版本不匹配。但按照上面流程找对应版本应该就能避免这个问题。