程序报错之torch、torch-geometric、torch_sparse等版本依赖问题

1 引言

复现代码的过程中遇到了

• 段错误 (核心已转储)
• AttributeError: Can’t get attribute ‘DataEdgeAttr’ on
• OSError: libcudart.so.11.0: cannot open shared object file: No such file or directory
  等等一系列错误。

经过查询发现可能原因是torch库与torch-scatter、torch-spare、torch-geometric版本不对应，torch-scatter、torch-spare、torch-geometric等包依赖于torch库。

2 解决方法（重新安装torch、torch-geometric库）

2.1 查询cuda版本，下载对应的torch库

在LINUX下输入cat /usr/local/cuda/version.txt，返回对应的cuda版本。

访问pytorch官网https://pytorch.org/
选择想要下载的torch版本，我选择的是1.7.0版本

pip install torch==1.7.0+cu101 torchvision==0.8.0+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

下载后，查看版本

>>> import torch
>>> print(torch.__version__)
# 输出：1.7.0+cu101

2.2 先下载对应版本的torch_scatter、torch_sparse等

https://pytorch-geometric.com/whl/
在以上链接找到对应torch及cuda版本
【一定要在此界面进入找，不然很容易出现不匹配问题！】

torch-1.7.0+cu101版本下对应的有
torch_cluster、torch_scatter、torch_sparse、torch_spline_conv四个与之版本依赖的库
cp38代表所对应的python版本为3.8，分别有对应的linux、windows版本。

我的python版本为3.8，torch-1.7.0+cu101，选择对应的torch_cluster和torch_sparse如上图标红框所示。

2.3 安装.whl文件

pip install 下载的whl文件所在路径
【最好是切换到下载好的.whl文件目录下直接pip install + 文件名】

2.4 安装 geometirc

安装好上面的四个包之后 直接pip install geometric即可安装geometric包

重新安装成功！不报错了！