Ubuntu 系统 cuda12.2 安装 MMDetection3D

DataBall 助力快速掌握数据集的信息和使用方式,会员享有 百种数据集,持续增加中。

需要更多数据资源和技术解决方案,知识星球: "DataBall - X 数据球(free)"

贵在坚持!


1、创建python虚拟环境

bash 复制代码
conda create -n mm3d python=3.8

2、进入python虚拟环境

bash 复制代码
conda activate mm3d

3、安装对应版本的 torch

bash 复制代码
pip install torch==2.1.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu122

4、安装 mmcv

bash 复制代码
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu122/torch2.1.0/index.html

5、安装 mmengine

bash 复制代码
pip install mmengine==0.9.0

6、安装 mmdet

bash 复制代码
pip install mmdet==3.2.0

7、克隆项目 mmdetection3d

bash 复制代码
git clone https://github.com/open-mmlab/mmdetection3d.git

8、进入项目文件夹

bash 复制代码
cd mmdetection3d/

9、编译安装依赖

bash 复制代码
pip install -r requirements/build.txt

10、安装 mmdetection3d

bash 复制代码
pip install -v -e .

11、安装工具 openmim

bash 复制代码
pip install -U openmim

12、下载预训练模型,并将模型保存在文件夹checkpoint

bash 复制代码
mim download mmdet3d --config pointpillars_hv_secfpn_8xb6-160e_kitti-3d-3class --dest checkpoint

13、测试 mmdet3d 安装是否成功

bash 复制代码
from mmdet3d.apis import init_model, inference_detector
print("build_detector imported successfully!")

如果程序执行正确,说明安装成功,准备好相应数据集,可以跑项目自带的demo示例。

助力快速掌握数据集的信息和使用方式。

数据可以如此美好!

相关推荐
Dongsheng_20192 小时前
【汽车篇】AI深度学习在汽车零部件外观检测——机电轴承的应用
人工智能·深度学习·汽车
en-route3 小时前
从零开始学神经网络——LSTM(长短期记忆网络)
人工智能·深度学习·lstm
DIY机器人工房4 小时前
NAT 模式、命令行版、桥接模式方式给ubuntu虚拟机配网步骤:
linux·网络协议·ubuntu·嵌入式·桥接模式·diy机器人工房
努力毕业的小土博^_^5 小时前
【深度学习|学习笔记】详细讲解一下 深度学习训练过程中 为什么 Momentum 可以加速训练?
人工智能·笔记·深度学习·学习·momentum
清风吹过5 小时前
少样本学习论文分享:多模态和类增量学习
论文阅读·人工智能·深度学习·学习·机器学习
什么半岛铁盒6 小时前
C++项目:仿muduo库高并发服务器---------LoopThreadPool模块和TcpServer模块的实现
linux·服务器·c++·mysql·ubuntu
无妄无望7 小时前
解码器系列(1)BERT
人工智能·深度学习·bert
葡萄与www7 小时前
模块化神经网络
人工智能·深度学习·神经网络·机器学习
colus_SEU7 小时前
【循环神经网络3】门控循环单元GRU详解
人工智能·rnn·深度学习·机器学习·gru
阿_旭7 小时前
基于深度学习的CT扫描图像肝脏肿瘤智能检测与分析系统【python源码+Pyqt5界面+数据集+训练代码】
人工智能·python·深度学习·肝脏肿瘤分割