文章目录
前言
本文主要说明Linux系统下配置mmdetection3d环境详细步骤。包括pytorch,cuda,mmcv,mmdet等的安装详细过程。
详细过程
1.创建虚拟环境
-
使用conda创建虚拟环境,并进入该环境。注意:你的系统必须要有annaconda;可以去annaconda官网自行下载你需要的版本。
在终端输入命令:
conda create -n mmdet3d python=3.7 -y
conda activate mmdet3d
2.安装PyTorch
- 可以在PyTorch官网,使用conda名进行下载安装。
- 最新版本的CUDA11.3以及PyTorch安装命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
3.安装MMCV
- 安装MMCV两种方式:
- 方式一: 查询基于自己安装cuda和pytorch版本的mmcv;使用下面命令安装预编译版本的mmcv-full:
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
- 需要把命令行中的 {cu_version} 和 {torch_version} 替换成对应的版本。例如:在 CUDA 11 和
PyTorch 1.7.0 的。具体版本在上面的查询链接可以直接得到。
方式二:直接用下面命令行进行安装
pip install mmcv-full
- 注意: 最好使用第一种方式,方式二可能会失败。
4.安装MMDetection
- 同样两种方式: 方式一:从源码编码MMDectction:
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
git checkout v2.19.0 # 转到 v2.19.0 分支
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
- 方式二:
pip install mmdet
**
- 注意:同MMCV,方式二可能会报错,最好采用方式一安装。
**
5.安装MMSegmentation
- 两种方式:
- 方式一:源码编译
git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
git checkout v0.20.0 # switch to v0.20.0 branch
pip install -e . # or "python setup.py develop"
- 方式二:
pip install mmsegmentation
6.安装依赖包和MMDetection3D
- 首先:克隆MMDetection3D代码
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
- 然后:安装包
pip install -v -e . # or "python setup.py develop"
作者安装过程出现的bug
- 安装mmdet的时候,出现pyccotools安装一直失败。这里说明一下,使用pip命令是不能成功的,使用命令:
conda install pycocotools -c Esri
- 即可成功安装,这一步可以在进行mmdet安装之前进行。