下载MUTAG数据集时报错:
from torch_geometric.datasets import TUDataset
dataset = TUDataset(root='data/TUDataset', name='MUTAG')
是由于 torch_geometric 版本太高导致的
如果直接 pip install torch_geometric,会下载到最新版本的torch_geometric(如2.5.3)
解决办法:
安装2.2.0版本的 torch_geometric
pip install torch_geometric==2.2.0
注意:低版本的 torch_geometric 依赖 torch-cluster、pytorch-scatter、torch-sparse、pytorch-spline-conv,还需安装以上四个库,可以参考博客:亲测!!安装相应版本的torch-geometric
数据集下载成功: