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

1 引言

复现代码的过程中遇到了

• 段错误 (核心已转储)
• AttributeError: Can’t get attribute ‘DataEdgeAttr’ on <module ‘torch_geometric.data.data’ from ‘/home/qijj/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch_geometric/data/data.py’>
• 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、torch-geometric

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

pip install torch-geometric 安装pyg
加镜像源安装更快：
pip install torch-geometric -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch-geometric -i https://mirrors.aliyun.com/pypi/simple/
pip install torch-geometric -i https://pypi.doubanio.com/simple

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

这篇好文章是转载于：学新通技术网