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

  1. https://github.com/open-mmlab/mmcv
  2. 简介
    • 是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 算子
  3. 版本
    • MMCV有两个版本:
    • mmcv-full全面,具有完整的功能和各种开箱即用的 CUDA操作。建造需要更长的时间。
    • mmcv精简版没有 CUDA 操作,但具有所有其他功能,类似于 mmcv<1.0.0。当您不需要那些 CUDA 操作时,它很有用。
  4. 安装
    • 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
  5. 注意事项
    • 注意:不要在同一个环境下安装两个版本,否则可能会遇到类似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,这个是一定可以配置成功的,其他的版本没试过,不做保证,可能会有各种各样的坑,如果愿意的话,可以自行尝试
  6. 报错
    • ERROR: Failed building wheel for mmcv-full
      • ERROR: Command errored out with exit status 1
      • 安装过程中报错,原因是没有C++环境!
      • windows端安装的时候,不能使用pip直接安装,需要使用源码安装
      • 其实我没有安装他的这个方法来做,而是直接使用了命令行来做,一下子就好了,也没有报错,但是后续还是有很多坑,所以最好是按照人家的这个做法来进行修改编译
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,367评论 6 512
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,959评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,750评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,226评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,252评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,975评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,592评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,497评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,027评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,147评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,274评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,953评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,623评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,143评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,260评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,607评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,271评论 2 358

推荐阅读更多精彩内容