Bootstrap

pip安装torch_scatter, torch_sparse报错

在安装torch_geometric时还需要安装torch_scatter和torch_sparse,但是直接使用pip install安装会报错,报错内容是缺乏依赖项。查了好久发现直接下载whl到本地再使用pip安装时最不容易出错的,这里记录一下,避免再次掉坑。

首先需要进入到python环境下,查看torch版本

import torch
torch.__version__

在python环境中运行上述代码后,会打印torch以及cuda的版本,例如,我的torch是2.0.1版本的,cuda是11.7版本的,打印结果就会是"2.0.1+cu117"。然后进入torch_geometric的whl网址https://pytorch-geometric.com/whl/,会发现有很多不同的torch版本,如下所示。
在这里插入图片描述
找到符号自己环境的版本,点击就会进入到torch_scatter和torch_sparse的下载地址。如下所示,
在这里插入图片描述
点击相应的版本即可下载相应的whl文件到本地。在命令行环境中,cd到whl文件所在的路径下,使用pip install whl文件名 即可成功安装。

;