结论
把 torch 版本由 cpu 版本改为兼容 cuda 的版本。
例如
把torch1.10.0
改为torch1.10.0+cu111
1 问题描述
在使用torch_geometric库时,出现如下报错
OSError Traceback (most recent call last)
Cell In[2], line 1
----> 1 from torch_geometric.datasets import KarateClub
3 dataset = KarateClub()
4 print(f'Dataset: {dataset}:')
File ~/miniconda3/envs/BESTIE/lib/python3.8/site-packages/torch_geometric/__init__.py:13
11 import torch_geometric.loader
12 import torch_geometric.transforms
---> 13 import torch_geometric.datasets
14 import torch_geometric.nn
15 import torch_geometric.explain
File ~/miniconda3/envs/BESTIE/lib/python3.8/site-packages/torch_geometric/datasets/__init__.py:101
99 from .sbm_dataset import RandomPartitionGraphDataset
100 from .mixhop_synthetic_dataset import MixHopSyntheticDataset
--> 101 from .explainer_dataset import ExplainerDataset
102 from .infection_dataset import InfectionDataset
103 from .ba2motif_dataset import BA2MotifDataset
File ~/miniconda3/envs/BESTIE/lib/python3.8/site-packages/torch_geometric/datasets/explainer_dataset.py:9
7 from torch_geometric.datasets.graph_generator import GraphGenerator
8 from torch_geometric.datasets.motif_generator import MotifGenerator
----> 9 from torch_geometric.explain import Explana