ImportError:libcudart.so.9.0:无法打开共享对象文件:没有这样的文件或目录(Ubuntu)

2024-10-02 08:27:17 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在python对象检测包mmdetection(在ubuntu 16.04上)中运行演示代码,但得到如下错误:

Traceback (most recent call last):
File "image_demo.py", line 3, in
from mmdet.apis import inference_detector, init_detector, show_result_pyplot
File "/home/hustwen/mmdetection/mmdet/apis/init.py", line 1, in
from .inference import (async_inference_detector, inference_detector,
File "/home/hustwen/mmdetection/mmdet/apis/inference.py", line 7, in
from mmcv.ops import RoIPool
File "/home/hustwen/anaconda3/envs/mmdet/lib/python3.7/site-packages/mmcv/ops/init.py", line 1, in
from .bbox import bbox_overlaps
File "/home/hustwen/anaconda3/envs/mmdet/lib/python3.7/site-packages/mmcv/ops/bbox.py", line 3, in
ext_module = ext_loader.load_ext('_ext', ['bbox_overlaps'])
File "/home/hustwen/anaconda3/envs/mmdet/lib/python3.7/site-packages/mmcv/utils/ext_loader.py", line 11, in load_ext
ext = importlib.import_module('mmcv.' + name)
File "/home/hustwen/anaconda3/envs/mmdet/lib/python3.7/importlib/init.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory

为了运行这个软件包,我升级了我的nvidia驱动程序和cuda(从cuda9.0升级到cuda10.1),并根据新的cuda版本按照说明安装pytorch、mmcv、mmdetection。但是仍然有这个错误,我认为这个错误与旧的cuda(cuda9.0)有关,但是无法找出错误在哪里

我查了libcuadrt,但只查到了libcudart

(mmdet) hustwen@TitanXP:~$ function lib_installed() { /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep $1; }
(mmdet) hustwen@TitanXP:~$ function check() { lib_installed $1 && echo "$1 is installed" || echo "ERROR: $1 is NOT installed"; }
(mmdet) hustwen@TitanXP:~$ check libcuda
    libcudart.so.10.1 -> libcudart.so.10.1.243
    libcuda.so.1 -> libcuda.so.430.64
    libcuda is installed
(mmdet) hustwen@TitanXP:~$ check libcudart
    libcudart.so.10.1 -> libcudart.so.10.1.243
    libcudart is installed

一些必要的信息:

pytorch 1.7.1 py3.7_cuda10.1.243_cudnn7.6.3_0 pytorch
torchaudio 0.7.2 py37 pytorch
torchvision 0.8.2 py37_cu101 pytorch

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

mmcv-full 1.2.1

mmdet version = '2.7.0'

gcc version 5.4.0 20160609 (Ubuntu 5.4.0-6ubuntu1~16.04.12)

Tags: installedinpyimporthomeliblinepytorch

热门问题