欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 汽车 > 维修 > Nvidia Jetson Orin平台部署CenterPoint模型

Nvidia Jetson Orin平台部署CenterPoint模型

2024/10/25 22:36:49 来源:https://blog.csdn.net/m0_48100146/article/details/142927959  浏览:    关键词:Nvidia Jetson Orin平台部署CenterPoint模型

最近尝试将CenterPoint模型部署到Orin平台,网络上教程很多,也很杂乱,于是便整理一版自用。

主要根据NVIDIA Lidar AI Solution进行复现。并在此基础上进行补充

Orin平台:
python:3.8
CUDA:11.4
torch:1.14.0
torchvision:0.15.1
TensorRT: 8.5.2.1

在Compile && Run(编译&&运行)阶段没有报错,但是他只给了两个bin文件,于是便想试一下Evaluate Inference Result(评估推理结果)部分
在这里插入图片描述

按上面的图片结构下载好nuscenes数据集(注意这里是nuScenes文件夹,后面CenterPoint模型的路径是nuscenes,注意更改路径)

一、下载centerpoint模型到CUDA-CenterPoint文件夹中,配好环境(重点安装cumm和spconv)

Jetson Orin 安装spconv问题

必须要先确定版本,我的版本:spconv == 2.1.21,cumm == 0.2.8, ccimport == 0.3.7

1、安装之前检查系统有无spconv和cumm,通过以下命令检查,如果有,则用pip卸载

pip list | grep spconvpip list | grep cumm

2、先下载cumm==0.2.8,进入虚拟环境执行以下命令

export CUMM_CUDA_VERSION="11.4" # 11.4为cuda版本
export CUMM_DISABLE_JIT="1" # 不用JIT编译cumm,而是编译成whl后再安装
export CUMM_CUDA_ARCH_LIST

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com