环境配置 | MMDetection3D v0.17.1

本文是配置 MMDetection3D v0.17.1 的笔记（该版本是 DAIR-V2X 代码依赖的指定版本），安装过程详见官方安装说明。我采用的是最简安装，如果有需要修改源码自定义功能的话，可以通过编译源码安装。

1. 创建虚拟环境

conda create -n open-mmlab python=3.7 -y

2. 进入虚拟环境，安装 pytorch

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c pytorch

版本 >1.3 即可，具体各版本的安装命令可查看 pytorch 官网，根据自己的 CUDA 版本选择对应的命令安装。CUDA 版本可以使用 nvcc -V 命令查看，我的 CUDA 版本是 10.2。

如果没有换过 conda 的源且安装速度极慢，可以参考 Anaconda添加清华镜像源 这篇博客换源，加速下载。

3. 安装 MMCV

去 官网 根据 CUDA 版本和 torch 版本获取 MMCV 的安装命令。pip 安装慢的话可以换源，最简单的方法是直接在命令后面加上清华源 -i https://pypi.tuna.tsinghua.edu.cn/simple

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

4. 安装 MMDetection

pip install mmdet==2.14.0

5. 安装 MMSegmentation

pip install mmsegmentation==0.14.1

6. 克隆 MMDetection3D 仓库

git clone https://github.com/open-mmlab/mmdetection3d.git

没开梯子克隆卡住的话，多取消重试几次就行

7. 进入仓库，安装 MMDetection3D

pip install -v -e .

至此，安装完毕。