Windows10下面安装MMCV全过程图文详解

https://github.com/open-mmlab/mmcv简介是OpenMMLab家自己开发的模块、包，其他框架的运行，例如MMDetection 的运行是依赖于这个库的MMCV 是一个面向计算机视觉的基础库，它支持了很多开源项目，例如：MMClassification: OpenMMLab 图像分类工具箱MMDetection: OpenMMLab 目标检测工具箱MMDetection

文章共2,290字 · 阅读需要大约8分钟

一键AI生成摘要，助你高效阅读

问答

我真的爱发明

8392人浏览 · 2021-10-30 22:50:37

我真的爱发明 · 2021-10-30 22:50:37 发布

https://github.com/open-mmlab/mmcv
简介
- 是OpenMMLab家自己开发的模块、包，其他框架的运行，例如MMDetection 的运行是依赖于这个库的
- MMCV 是一个面向计算机视觉的基础库，它支持了很多开源项目，例如：
  - MMClassification: OpenMMLab 图像分类工具箱
  - MMDetection: OpenMMLab 目标检测工具箱
  - MMDetection3D: OpenMMLab 新一代通用 3D 目标检测平台
  - MMSegmentation: OpenMMLab 语义分割工具箱
  - MMAction2: OpenMMLab 新一代视频理解工具箱
  - MMTracking: OpenMMLab 一体化视频目标感知平台
  - MMPose: OpenMMLab 姿态估计工具箱
  - MMEditing: OpenMMLab 图像视频编辑工具箱
  - MMOCR: OpenMMLab 全流程文字检测识别理解工具包
  - MMGeneration: OpenMMLab 图片视频生成模型工具箱
- MMCV 提供了如下众多功能：
  - 通用的 IO 接口
  - 图像和视频处理
  - 图像和标注结果可视化
  - 常用小工具（进度条，计时器等）
  - 基于 PyTorch 的通用训练框架
  - 多种 CNN 网络结构
  - 高质量实现的常见 CUDA 算子
版本
- MMCV有两个版本：
- mmcv-full：全面，具有完整的功能和各种开箱即用的 CUDA操作。建造需要更长的时间。
- mmcv：精简版，没有 CUDA 操作，但具有所有其他功能，类似于 mmcv<1.0.0。当您不需要那些 CUDA 操作时，它很有用。
安装
- Linux 下pip安装
  - MMCV的版本需要和PyTorch以及cuda的版本匹配才行
    - https://github.com/open-mmlab/mmcv#installation
    - 不是每一个cuda和PyTorch版本，MMCV都是支持的，只是支持其中的一部分
    - 这里我喜欢安装最新版的
    - pip install mmcv-full==1.3.16 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
- Windows下源码安装
  - B站教程
    - Win10配置MMSegmentation并训练自己数据集_哔哩哔哩_bilibili
    - 下载代码
      - 这个check out是选择自己的版本
      - git clone https://github.com/open-mmlab/mmcv.git cd mmcv git checkout v1.3.13 pip install -r requirements.txt
    - 编译
      - 修改visual studio中的配置文件
        
        用Everything找到2019的x86和x61下面的cl.exe
        
        自己的路径是
        
        C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\VC\Tools\MSVC\14.20.27508\bin\Hostx86\x64
      - 修改环境变量
        
        然后将上面这个路径添加到自己系统的环境变量中，并且将路径上移到最顶层
        
        同时将CUDA_PATH 里面cuda的版本设置为10.2
        
        在命令行中输入set Path=C可以让环境变量的修改立马生效
        
        保存之后重新打开命令行
        
        输入cl，若是显示出了19的编译器，那么就证明没有问题了
      - 使用Anaconda的powershell进行编译
        
        这个一定是用使用Anaconda的powershell进行安装的，使用其他命令行是会报错的
        
        其他的非powershell会报下面的错误
        
        而Anaconda下面输入对应的命令则不会报这个错误
        
        切换环境
        
        单独一个activate不好使，前面必须加一个conda
        
        conda activate MMDetection
        
        pip install -r requirements.txt 安装MMCV需要的包
        
        根据显卡算力不同设置对应的参数
        
        查看英伟达显卡算力
        
        这里可以看到，我们的RTX2060的显卡算力是7.5的
        
        设置对应的参数
        
        $env:TORCH_CUDA_ARCH_LIST="7.5"
        
        $env:MMCV_WITH_OPS = 1
        
        $env:MAX_JOBS = 8
        
        # 根据你可用CPU以及内存量进行设置
        
        电脑一般的话，默认为4，若是电脑配置比较高的话，可以设置为8，其他数字就不确定了，一句话，4是肯定没有问题的，其他的值自行试验
        
        这里需要强调一点儿，就是必须用Anaconda 的powershell才行，否则会报文件名、目录名或卷标语法不正确。
        
        将上述参数输入到命令行中
      - 进行编译安装
        
        python setup.py build_ext
        
        python setup.py build_ext
        
        会显示下面的界面，有一个过程，这个才算在编译，若是一闪而过的话，那么大概率是出问题了
        
        编译完成后的界面长下面这样
        
        python setup.py develop
        
        python setup.py develop
        
        完成后长下面这样
      - 编译报错的解决办法
        
        mmcv在windows上的安装踩坑记 - 知乎
        
        其实我没有安装他的这个方法来做，而是直接使用了命令行来做，一下子就好了，也没有报错，但是后续还是有很多坑，所以最好是按照人家的这个做法来进行修改编译
  - 官方教程
    - https://mmcv.readthedocs.io/en/latest/get_started/build.html
注意事项
- 注意：不要在同一个环境下安装两个版本，否则可能会遇到类似ModuleNotFoundError. 您需要先卸载一个，然后再安装另一个。Installing the full version is highly recommended if CUDA is available.
- windows端安装的时候，不能使用pip直接安装，需要使用源码安装
- 注意：MMCV 需要 Python3.6++。
- 编译的时候不能使用CMD进行编译，必须使用powershell进行编译才行
- cuda就用10.2 PyTorch就用1.8，这个是一定可以配置成功的，其他的版本没试过，不做保证，可能会有各种各样的坑，如果愿意的话，可以自行尝试
报错
- ERROR: Failed building wheel for mmcv-full
  - ERROR: Command errored out with exit status 1
  - 安装过程中报错，原因是没有C++环境！
  - windows端安装的时候，不能使用pip直接安装，需要使用源码安装
  - 其实我没有安装他的这个方法来做，而是直接使用了命令行来做，一下子就好了，也没有报错，但是后续还是有很多坑，所以最好是按照人家的这个做法来进行修改编译