ImportError: libcudart.so.10.0:cannot open shared object file: No such file or direct【mmdetection错误】

问题:在使用mmdetection做训练的时候,出现错误
ImportError: libcudart.so.10.0:cannot open shared object file: No such file or direct【mmdetection错误】_第1张图片

>>>from mmdet.apis import init_detector
Traceback (most recent call last):
  *****
  return __bootstrap.gcd_import(name[level:],package, level)
ImportError: libcudart.so.10.0:cannot open shared object file: No such file or directory

在官网的issue中都认为是cuda的版本问题,但看报错都是mmcv的文件加载不进来,盲猜是mmcv的版本出问题了。

环境版本
ubuntu16.04+Anaconda3+python3.7.7+cuda10.0+cuDNN7.6.4.3
pytorch1.6.0+python3.7

解决办法:
mmcv的安装用下面这句:

pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html

mmdetection的安装中mmcv是最容易出现问题的一个地方,经常是因为mmcv版本不匹配导致安装不能正确进行。使用

pip install mmcv-full

并不一定好用,要使用上述指定版本号的安装方式。

你可能感兴趣的:(mmdetection,cuda,python,深度学习,mmdetection,mmcv)