- 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
-
MMCV的版本需要和PyTorch以及cuda的版本匹配才行
-
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
-
用Everything找到2019的x86和x61下面的cl.exe
-
修改环境变量
-
然后将上面这个路径添加到自己系统的环境变量中,并且将路径上移到最顶层
-
-
同时将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才行,否则会报文件名、目录名或卷标语法不正确。
-
查看英伟达显卡算力
-
将上述参数输入到命令行中
-
-
这个一定是用使用Anaconda的powershell进行安装的,使用其他命令行是会报错的
-
进行编译安装
-
python setup.py build_ext
- python setup.py build_ext
- 会显示下面的界面,有一个过程,这个才算在编译,若是一闪而过的话,那么大概率是出问题了
-
- 编译完成后的界面长下面这样
-
-
python setup.py develop
- python setup.py develop
-
- 完成后长下面这样
-
-
python setup.py build_ext
-
编译报错的解决办法
- mmcv在windows上的安装踩坑记 - 知乎
- 其实我没有安装他的这个方法来做,而是直接使用了命令行来做,一下子就好了,也没有报错,但是后续还是有很多坑,所以最好是按照人家的这个做法来进行修改编译
-
-
修改visual studio中的配置文件
-
官方教程
- https://mmcv.readthedocs.io/en/latest/get_started/build.html
-
B站教程
-
Linux 下pip安装
-
注意事项
- 注意:不要在同一个环境下安装两个版本,否则可能会遇到类似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直接安装,需要使用源码安装
- 其实我没有安装他的这个方法来做,而是直接使用了命令行来做,一下子就好了,也没有报错,但是后续还是有很多坑,所以最好是按照人家的这个做法来进行修改编译
-
-
ERROR: Failed building wheel for mmcv-full
Windows10下面安装MMCV全过程图文详解
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 最近本人在跑Flownet2代码,从安装到运行遇到了蛮多问题,记录一下安装过程。 一、工具篇 1. Flownet...
- 5.3 Windows安装说明 译者:Python 文档协作翻译小组,原文:Windows Installatio...
- 关于TVM在Windows上的安装教程网上非常少,原因是TVM主要是针对Linux开发的,目前在Windows运行...