解决OSError: libtorch_cuda_cpp.so: cannot open shared object file: No such file or directory

最新推荐文章于 2025-02-28 20:39:04 发布

原创

最新推荐文章于 2025-02-28 20:39:04 发布 · 3.7k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #pytorch #python

文章目录

结论
1 问题描述
2 解决过程
3 其他

结论

把 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.explai

最低0.47元/天解锁文章