Linux系统配置mmdetection3d环境

文章目录

  • 前言
  • 详细过程
    • 1.创建虚拟环境
    • 2.安装PyTorch
    • 3.安装MMCV
    • 4.安装MMDetection
    • 5.安装MMSegmentation
    • 6.安装依赖包和MMDetection3D
  • 作者安装过程出现的bug


前言

本文主要说明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名进行下载安装。

Linux系统配置mmdetection3d环境_第1张图片

  • 最新版本的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安装之前进行。

你可能感兴趣的:(MMDetection3D,深度学习,linux,pytorch)