MMDetection3D库的安装指导

calendar Oct 30, 2023 · 1 min read  ·
Share on: linkedin copy

Overview

准备

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

Miniconda

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

Linux下安装：

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

初始化：

1~/miniconda3/bin/conda init bash
2~/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：

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

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

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

安装依赖

创建一个虚拟环境：

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

安装PyTorch v1.13.1：

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

安装SpConv：

1pip install spconv-cu116

安装MMDetection3D

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

安装MM相关依赖

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

安装MMDetection3D库

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

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

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

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

测试

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

1from mmdet3d.apis import init_model, inference_detector

或者

1import mmdet3d
2print(mmdet3d.__version__)

comments powered by Disqus