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

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

数据可以如此美好!

相关推荐
烟锁池塘柳03 分钟前
【大模型】解码策略:Greedy Search、Beam Search、Top-k/Top-p、Temperature Sampling等
人工智能·深度学习·机器学习
zzc9211 小时前
时频图数据集更正程序,去除坐标轴白边及调整对应的标签值
人工智能·深度学习·数据集·标签·时频图·更正·白边
Blossom.1182 小时前
机器学习在智能供应链中的应用:需求预测与物流优化
人工智能·深度学习·神经网络·机器学习·计算机视觉·机器人·语音识别
Gyoku Mint2 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类
m0_751336394 小时前
突破性进展:超短等离子体脉冲实现单电子量子干涉,为飞行量子比特奠定基础
人工智能·深度学习·量子计算·材料科学·光子器件·光子学·无线电电子
有Li8 小时前
通过具有一致性嵌入的大语言模型实现端到端乳腺癌放射治疗计划制定|文献速递-最新论文分享
论文阅读·深度学习·分类·医学生
张较瘦_11 小时前
[论文阅读] 人工智能 | 深度学习系统崩溃恢复新方案:DaiFu框架的原位修复技术
论文阅读·人工智能·深度学习
cver12311 小时前
野生动物检测数据集介绍-5,138张图片 野生动物保护监测 智能狩猎相机系统 生态研究与调查
人工智能·pytorch·深度学习·目标检测·计算机视觉·目标跟踪
学技术的大胜嗷11 小时前
离线迁移 Conda 环境到 Windows 服务器:用 conda-pack 摆脱硬路径限制
人工智能·深度学习·yolo·目标检测·机器学习
好奇的菜鸟12 小时前
如何在 Ubuntu 24.04 (Noble) 上使用阿里源
linux·运维·ubuntu