pyG框架报错: OSError: libcudart.so.10.1: cannot open shared object file: No such file or directory

  继续学习pyG (Pytorch Geometric)框架,今天突发报了这个错——OSError: libcudart.so.10.1: cannot open shared object file: No such file or directory,如下图所示。

pyG框架报错: OSError: libcudart.so.10.1: cannot open shared object file: No such file or directory_第1张图片

报错提示

  根据这个报错提示,错误发生在’import torch_geometric as tg’的时候,之后一连串都是框架的源码。此时,又留意到报错提示里包含的’OSError: libcudart.so.10.1’中的’10.1’和我安装的torch_geometric时所选择的cuda版本一致,所以我猜测是环境安装问题。然后,我用’torch.cuda.is_available()'命令测试(具体步骤参考我的博文),果然Pytorch调用显卡失败,印证了我的猜想。紧接着,我查看设备的cuda版本(具体步骤参考我的博文和另一篇博文),发现此时所用的设备cuda版本为10.2,证实是环境安装问题。

  当按照cuda版本为10.2进行Pytorch和Pytorch Geometric框架安装后,报错消失。因此,当问题出现在import环节的时候,要仔细检查环境和包的安装问题。

你可能感兴趣的:(深度学习,python,cuda,pytorch,神经网络)