【MMdetection3d】Step1:环境搭建

Step1:环境搭建

  • 1.创建并激活虚拟环境
    • 1.1 用官方Pytorch指令安装!
    • 1.2 用官方mmcv指令安装!
  • 2 安装MMDetection
  • 3 克隆编译mmdetection3d
  • 4 环境测试
  • 5 测试demo

在Conda虚拟环境中搭建MMdetection3d环境

1.创建并激活虚拟环境

conda create -n mm3d python=3.9 -y
conda activate mm3d

1.1 用官方Pytorch指令安装!

Pytorch官网

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
#  或者
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia
# 查看torch的版本
python -c 'import torch;print(torch.__version__)'
python -c 'import torch;print(torch.cuda.is_available())'

1.2 用官方mmcv指令安装!

mmcv官网

# mim是OpenMMLab项目的包管理工具,它使mmcv的安装变得容易
pip install -U openmim
mim install mmcv
# 测试
python -c 'import mmcv'

2 安装MMDetection

参考官网链接:BUILD MMCV FROM SOURCE

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/optional.txt
nvcc --version
gcc --version
pip install -v -e .  # or "python setup.py develop"
# 测试
python -c 'import mmdet'

3 克隆编译mmdetection3d

git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip install -r requirements/optional.txt
pip install -v -e .  # or "python setup.py develop"
# 测试
python -c 'import mmdet3d'

4 环境测试

(mm3d) ll@ll:/media/ll/L/llr/a2023_my_3d/mmdetection3d$ python
Python 3.9.17 (main, Jul  5 2023, 20:41:20) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> import mmcv
>>> import mmdet
>>> import mmdet3d
>>> 

5 测试demo

1、下载预训练模型:hv_second_secfpn_6x8_80e_kitti-3d-car_20200620_230238-393f000c.pth

2、测试命令:

python demo/pcd_demo.py demo/data/kitti/000008.bin configs/3dssd/3dssd_4xb4_kitti-3d-car.py my_checkpoints/hv_second_secfpn_6x8_80e_kitti-3d-car_20200620_230238-393f000c.pth

【MMdetection3d】Step1:环境搭建_第1张图片

你可能感兴趣的:(python,笔记)