cudnn安装和mmcv安装过程中的问题

在网站中找到和cuda版本相应的cudnn版本

https://developer.nvidia.cn/rdp/cudnn-archive#a-collapse805-101

解压文件

tar -xvf cudnn-10.1-linux-x64-v8.0.5.39.tgz

把文件进行拷贝

sudo cp cuda/include/cudnn* /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看cudnn版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

在运行maskrcnn时 一直提示mmcv问题 应该安装mmcv-full

在网站https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html中查找到适合当前cuda torch版本的mmcv进行下载

pip install mmcv-full==1.5.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7/index.html