MMCV是一个开源的计算机视觉库,广泛用于基于Pytorch的深度学习项目中。本教程主要介绍MMCV的安装方法。
首先需要确认已安装Pytorch,CUDA及驱动版本。然后pip安装mmcv或从源码编译安装。需要注意 PyTorch、CUDA和MMCV版本匹配问题。
安装验证可以import mmcv测试是否成功。常见问题包括CUDA版本不匹配、编译错误等。可检查环境变量设置，删除后重新编译安装。
MMCV 是一个面向计算机视觉的基础库，它提供了以下功能：
- 图像和视频处理
- 图像和标注结果可视化
- 图像变换
- 多种 CNN 网络结构
- 高质量实现的常见 CUDA 算子
它这里支持很多的 OpenMMLab 项目，mmcv与pytorch和cuda有版本兼容的问题，你可以通过文档查看。安装 MMCV — mmcv 2.0.1 文档
目前网页打开是2.0.x版本，你可以通过下面方式去找到相应的版本：
在安装 mmcv 之前，请确保 PyTorch 已经成功安装在环境中，可以参考 PyTorch 官方安装文档。也可使用以下命令验证：
(torch1.8) C:\Users\asus>python
Python 3.8.17 (default, Jul 5 2023, 20:44:21) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.0.1+cu118
>>> print(torch.version.cuda)
11.8
>>>- python
- import torch
- print(torch.__version__)
- print(torch.version.cuda)
这里我们就使用pip安装即可，选择合适的系统，对应的cuda、pytorch版本，以及MMCV的版本选择。即可得到安装命令
需要注意的是，PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的，故 mmcv 只提供 1.x.0 的编译包。如果你 的 PyTorch 版本是 1.x.1，你可以放心地安装在 1.x.0 版本编译的 mmcv。例如，如果你的 PyTorch 版本是 1.8.1，你可以放心选择 1.8.x。
cuda118、pytroch2.0，mmcv2.0.0安装指令：
pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html如果你安装依赖库的时间过长，可以指定 pypi 源：
pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html -i https://pypi.tuna.tsinghua.edu.cn/simple