本文主要是介绍【技术】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__)
有问题欢迎留言，我会回复或更新文章哦，谢谢！
这篇关于【技术】MMDetection3D库的安装步骤记录的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
