windows环境安装mmdetection
- 创建pytorch环境
- 最终安装的版本信息
- 安装过程
- step1:安装mmcv-full
- step2:安装mmdetection
创建pytorch环境
mmdetection是基于PyTorch、CUDA环境来安装的,所以首先得安装深度学习pytorch环境、参考我的这篇pytorch环境安装文档
这里,我拷贝pt17环境为mmdetection
最终安装的版本信息
显卡:RTX 2080 ti
系统:Windows 10
cuda:11.0
cudnn:8.0.5.39
pytorch:1.7.1
torchvision:0.8.2
mmcv-full:
mmdetection:
查看cuda版本号:在cmd中键入:nvcc --version即可
安装过程
注意:以下安装都是在具体某一虚拟环境下的! 所以记得先在cmd中激活环境进入后再安装(我是安装在mmdetection虚拟环境下的)
step1:安装mmcv-full
mmcv是mm系列的底层支持,包括安装mmclassification、mmsegmentation都会使用到mmcv,这里有两个安装版本,一个是简化的版本mmcv,一个是完整的版本mmcv-full,完整的版本中包含了大量的算子,GPU下最好还是安装这个完整的版本,安装命令如下:
pip install mmcv==2.0.0rc3 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8/index.html
版本参考文献
Failed building wheel for mmcv-full
注意: 如果已经安装了 mmcv,首先需要使用 pip uninstall mmcv 卸载已安装的 mmcv,如果同时安装了 mmcv 和 mmcv-full,将会报 ModuleNotFoundError 错误。
step2:安装mmdetection
参考文章