Ubuntu18.04实现bmask-rcnn代码

Ubuntu18.04实现bmask-rcnn

配置环境

cuda 11.1在这里插入代码片
cudnn
cudatoolkit 11.1.1
python 3.8 刚开始使用的是3.6，但是在安装torch1.9版本的时候冲突，需要python3.7及以上，所以直接安装了3.8
torch 1.9.0+cu111
torchaudio 0.9.0
torchvision 0.10.0+cu111
（选用 torch1.9是因为torch1.8和bmask rcnn使用的框架detectron2冲突，会出现错误RuntimeError: radix_sort: failed on 1st step: cudaErrorInvalidDevice: invalid device ordinal）

// 创建环境
conda create --name detectron2 python==3.8
//下载cudatoolkit，在base环境中安装过了cuda和cudnn,这里没有安装也可使用
conda activate detectron2
conda install -c anaconda cudatoolkit=11.1.1
//安装totch等
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
//安装detectron2
 pip install fvcore
 pip install cython
 pip install pycocotool