jetson orin安装spconv问题

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

本文参考NVIDIA Jetson AGX Orin配置OpenPCDet环境部署PointPillar，和CenterPoint系列安装时遇到的问题

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

pip list | grep spconv

pip 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="8.7" # xavier是7.2，TX2是6.2，orin是8.7
git clone -b v0.2.8 https://github.com/FindDefinition/cumm # v0.2.8为cumm对应tag版本
cd cumm # cd到cumm的代码根目录
python setup.py bdist_wheel # 编译生成cumm的whl在dist文件夹内
pip install dists/xxx.whl # 安装编译好的cumm的whl，名字应该类似cumm_cu114-0.2.8-cp38-cp38m-linux_aarch64.whl

最后在python中导入cumm检查是否安装成功

3、下载spconv==2.1.21，找到文件下的pyproject.toml并删除cumm的版本要求，执行以下命令

export CUMM_CUDA_VERSION="11.4" # 11.4为cuda版本
export SPCONV_DISABLE_JIT="1" # 不用JIT编译spconv，而是编译成whl后再安装
export CUMM_CUDA_ARCH_LIST="8.7" # xavier是7.2，TX2是6.2，orin是8.7
git clone -b v2.1.22 https://github.com/traveller59/spconv --recursive # v2.1.22换成你想要用的github代码的spconv对应tag版本，注意需要加recursive
cd spconv # cd到spconv的代码根目录
pip install pccm wheel # 安装一些依赖包
python setup.py bdist_wheel # 编译生成spconv的whl在dist文件夹内
pip install dist/xxx.whl # 安装编译好的spconv的whl，名字应该类似spconv_cu114-2.1.22-cp38-cp38m-linux_aarch64.whl

在python中导入spconv检查是否成功。

如果遇到编译错误以及其他问题可以参考解决办法