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示例。

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

数据可以如此美好!

相关推荐
码字的字节7 小时前
深度学习损失函数的设计哲学:从交叉熵到Huber损失的深入探索
深度学习·交叉熵·huber
凪卄12138 小时前
图像预处理 二
人工智能·python·深度学习·计算机视觉·pycharm
碳酸的唐8 小时前
Inception网络架构:深度学习视觉模型的里程碑
网络·深度学习·架构
AI赋能8 小时前
自动驾驶训练-tub详解
人工智能·深度学习·自动驾驶
seasonsyy8 小时前
1.安装anaconda详细步骤(含安装截图)
python·深度学习·环境配置
deephub8 小时前
AI代理性能提升实战:LangChain+LangGraph内存管理与上下文优化完整指南
人工智能·深度学习·神经网络·langchain·大语言模型·rag
风吹落叶花飘荡9 小时前
Ubuntu系统 系统盘和数据盘扩容具体操作
linux·运维·ubuntu
go54631584659 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
Blossom.1189 小时前
基于深度学习的图像分类:使用Capsule Networks实现高效分类
人工智能·python·深度学习·神经网络·机器学习·分类·数据挖掘
宇称不守恒4.09 小时前
2025暑期—05神经网络-卷积神经网络
深度学习·神经网络·cnn