【技术】MMDetection3D库的安装步骤记录

最近折腾使用了MMDetection3D库，把安装的过程和环境配置都详细记录了下来，以方便人们避坑。blog
因为一不留神某个软件或驱动版本之间不匹配就会出现一些奇怪问题。

准备

本安装在以下环境中进行了测试：Ubuntu 20.04、CUDA-11.6、PyTorch v1.13.1。

Miniconda

此安装在conda虚拟环境中进行，因此请确保已安装Miniconda。

Linux下安装：

mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh

初始化：

~/miniconda3/bin/conda init bash
~/miniconda3/bin/conda init zsh

CUDA Toolkit

建议使用CUDA 11.6和Nvidia驱动程序510.39.01。Nvidia驱动程序可以与CUDA一起安装。

• 检查CUDA版本：nvcc --version
• 从系统中删除CUDA：sudo /usr/local/cuda-11.x/bin/cuda-uninstaller（将x替换为CUDA的版本）
• 检查Nvidia驱动程序版本：nvidia-smi
• 删除Nvidia驱动程序：sudo /usr/bin/nvidia-uninstall

安装CUDA11.6和驱动程序510.39.01：

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

在终端中勾选Nvidia驱动程序。

若需要安装其他版本，请查看官方网站。

安装依赖

创建一个虚拟环境：

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

安装PyTorch v1.13.1：

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

安装SpConv：

pip install spconv-cu116

安装MMDetection3D

了解更多信息：https://mmdetection3d.readthedocs.io/en/latest/get_started.html

安装MM相关依赖

# 安装mim
pip install -U openmim
mim install mmengine 'mmcv>=2.0.0rc4' 'mmdet>=3.0.0'

安装MMDetection3D库

情况一: 使用该库作为一个第三方包：

# 安装mmdet3d
mim install "mmdet3d>=1.1.0"

情况二：以源码方式安装：

git clone https://github.com/open-mmlab/mmdetection3d.git -b dev-1.x
cd mmdetection3d
pip install -v -e .

测试

要测试结果，请在Python环境中尝试从中导入库，例如：

from mmdet3d.apis import init_model, inference_detector

或者

import mmdet3d
print(mmdet3d.__version__)

有问题欢迎留言，我会回复或更新文章哦，谢谢！