DEBUG | mmdetection报错ImportError: libcudart.so.11.0: cannot open shared object file

1. 报错说明

安装完mmdetection之后，运行测试案例，报错

ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory

我搜了一下，看到有些帖子，是 libcudart.so.10.1，libcudart.so.9.0等。

2. 解决办法

我是重装了pytorch（按照cuda==11.0）安装的，然后重新build mmdetection（pip install -v -e .）然后就好了。

（步骤1）安装相应cuda版本的pytorch；

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

（步骤2）重装或者重新build mmdetection；

pip install -v -e .  # or "python setup.py develop"

3. 原因分析

我也不太确定，我感觉原因可能是 “与pytorch有关的cuda和与mmcv有关的cuda版本不一致导致的”。

需要说明的是，对于ubuntu系统中cuda版本的获取，有nvcc -V 和 nvidia-smi两种常用的办法。同时如果安装了pytorch-cuda，可以在python环境中，用 torch.version.cuda 来查看安装的torch相应的cuda版本。

补充资料：nvidia-smi 和 nvcc 结果的版本为何不一致。