安装PYG的前提,安装torch-scatter, torch-sparse, torch-cluster报错

安装torch-geometric之前需要安装torch-scatter, torch-sparse, torch-cluster

首先要保证安装的torch, torchvision, cuda的版本都相对应;

再安装torch_scatter,

然而,使用pip install安装torch-scatter报错了,但是使用指令:

pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.12.0+cu102.html

后安装就不会报错,添加了 --no-index;

安装的时候注意把上述链接中torch和cuda版本换成自己的。

同样将torch-scatter换成要安装的包名(torch-sparse, torch-cluster)

之后若要安装PYG 就直接

pip install torch-geometric

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