LINUX 安装 MMCV MMSegmentation

使用conda安装，cuda==11.3

1. 首先，使用conda生成openmmlab环境。

conda create --name openmmlab python=3.8 -y
conda activate openmmlab

 2. 在 https://pytorch.org/get-started/previous-versions 中找到自己对应的torch版本。

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

3. Installation — mmcv 2.0.0 documentation找到对应的mmcv版本 use pip:

pip install mmcv==2.0.0rc4 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.12/index.html

4.安装mmseg

git clone -b main https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .
# '-v' means verbose, or more output
# '-e' means installing a project in editable mode,
# thus any local modifications made to the code will take effect without reinstallation.

5.测试，首先安装mim

pip install -U openmim

其次测试结果：

mim download mmsegmentation --config pspnet_r50-d8_4xb2-40k_cityscapes-512x1024 --dest .
python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_4xb2-40k_cityscapes-512x1024.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --out-file result.jpg

结果如图所示： 

---

报错1：ImportError: /xxxx/mmcv/_ext.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe

原因：torch、cuda、mmcv版本不对应

报错2：AssertionError: bias_act miss in module _ext

原因：mmcv、mmseg等版本不对应

Error3:

SegneXt:

ImportError: cannot import name 'revert_sync_batchnorm' from 'mmcv.cnn.utils' (/root/anaconda3/envs/openmmlab/lib/python3.8/site-packages/mmcv/cnn/utils/__init__.py)

Reason and solution: https://download.openmmlab.com/mmcv/dist/cu113/torch1.12.0/index.html

find mmcv==1.6.0 and download:

pip install mmcv_full-1.6.0-cp38-cp38-manylinux1_x86_64.whl 

# 1.克隆代码仓库
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv

# 2.安装 ninja 和 psutil 以加快编译速度
pip install -r requirements/optional.txt

# 3.检查 nvcc 的版本（要求大于等于 9.2，如果没有 GPU，可以跳过）
nvcc --version

# 4.检查 gcc 的版本（要求大于等于5.4）
gcc --version

# 5.开始编译（预估耗时 10 分钟）
pip install -e . -v

# 6.验证安装
python .dev_scripts/check_installation.py

Output:

......
TorchVision: 0.13.0+cu113
OpenCV: 4.7.0
MMEngine: 0.7.3
MMCV: 2.0.0
MMCV Compiler: GCC 9.4
MMCV CUDA Compiler: 11.4
......