OpenMMLab系列安装教程

-我叫刘小小-

已于 2024-11-05 18:01:14 修改

阅读量661

点赞数 10

文章标签：深度学习人工智能 python3.11 pip conda

于 2024-11-05 14:54:31 首次发布

本文链接：https://blog.csdn.net/qq_45402302/article/details/143504240

版权

OpenMMLab系列安装教程（持续更新）

MMRotate

Github: https://github.com/open-mmlab/mmrotate?tab=readme-ov-file
Install README: https://mmrotate.readthedocs.io/en/latest/install.html

官方安装流程：

conda create --name openmmlab python=3.8 -y
conda activate openmmlab
conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch
pip install -U openmim
mim install mmcv-full
mim install mmdet\<3.0.0 （Windows：mim install mmdet<3.0.0 ）
pip install -v -e .

踩坑点：
首先要检查自己电脑的NVIDIA驱动版本，例如CMD运行nvidia-smi

直接安装cudatoolkit=10.2，由于驱动版本高，cudatoolkit版本太低，运行代码会报如下错误：

pytorch capability sm_86 is not compatible with the current PyTorch installation......

尝试了以下几个PyTorch和CUDA Toolkit版本后，得出结论：OpenMMLab系列工程与PyTorch版本似乎没有严格要求。因此，可以不按README里的PyTorch版本安装，其他（mmcv, mmdet）最好按REDME里的来安装。

Failed: conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge
Failed: conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge
Successful: conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

如果安装过程慢的扣脚，可以临时使用清华源，例如

mim install mmcv-full -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

MMSegmentation

Github: https://github.com/open-mmlab/mmsegmentation/tree/main
Install README: https://github.com/open-mmlab/mmsegmentation/blob/main/docs/en/get_started.md#installation
MMCV： https://mmcv.readthedocs.io/zh-cn/2.x/get_started/installation.html

官方安装流程：

conda create --name openmmlab python=3.8 -y
conda activate openmmlab
conda install pytorch torchvision -c pytorch
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
pip install -v -e .

踩坑点：
按照官方安装流程安装后，运行代码报错：

AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv>=2.0.0rc4.

点击进入assert代码段（L61）发现mmcv_min_version <= mmcv_version < mmcv_max_version，注意，右边是小于<而不是小于等于<=。然而，直接运行官方README里写的mim install "mmcv>=2.0.0"会安装最新版mmcv==2.2.0。
在这里插入图片描述
因此，需要自己手动卸载mmcv==2.2.0，然后安装区间内的mmcv版本（2.0.0rc4 <= mmcv_version < 2.2.0即可，这里安装了2.1.0）。

mim uninstall mmcv
mim install mmcv==2.1.0 -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple