torch_scatter安装

图卷积中常用的一个函数,直接pip install下载不下来,查阅各种资料之后跑通的一个比较方便的方法,记录一下:

1. 查看torch以及cuda版本:

import torch
print(torch.__version__)
print(torch.version.cuda)

torch_scatter安装_第1张图片

2. 找版本都对应上的torch_scatter并下载:

https://pytorch-geometric.com/whl/

torch_scatter安装_第2张图片 

 

 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] 找不到指定的程序。

说明还是下载的版本不匹配。但按照上面流程找对应版本应该就能避免这个问题。

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