主页 > 创业  > 

Ubuntu系统cuda12.2安装MMDetection3D

Ubuntu系统cuda12.2安装MMDetection3D

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

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

贵在坚持!

-----------------------------------------------------------------------------------------------

1、创建python虚拟环境

conda create -n mm3d python=3.8

2、进入python虚拟环境

conda activate mm3d

3、安装对应版本的 torch

pip install torch==2.1.0 torchvision torchaudio --extra-index-url download.pytorch.org/whl/cu122

4、安装 mmcv

pip install mmcv==2.1.0 -f download.openmmlab /mmcv/dist/cu122/torch2.1.0/index.html

5、安装 mmengine

pip install mmengine==0.9.0

6、安装 mmdet

pip install mmdet==3.2.0

7、克隆项目 mmdetection3d

git clone github /open-mmlab/mmdetection3d.git

8、进入项目文件夹

cd mmdetection3d/

9、编译安装依赖

pip install -r requirements/build.txt

10、安装 mmdetection3d

pip install -v -e .

11、安装工具 openmim

pip install -U openmim

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

mim download mmdet3d --config pointpillars_hv_secfpn_8xb6-160e_kitti-3d-3class --dest checkpoint

13、测试 mmdet3d 安装是否成功

from mmdet3d.apis import init_model, inference_detector print("build_detector imported successfully!")

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

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

数据可以如此美好!

标签:

Ubuntu系统cuda12.2安装MMDetection3D由讯客互联创业栏目发布,感谢您对讯客互联的认可,以及对我们原创作品以及文章的青睐,非常欢迎各位朋友分享到个人网站或者朋友圈,但转载请说明文章出处“Ubuntu系统cuda12.2安装MMDetection3D