关于激光雷达语义分割模型SPVNAS的配置流程

引言

其实在今年56月份时，我就配置好了相关环境，中间也遇到了不少坑。尤其是不小心弄串了环境把原来的环境配置给破坏了。防止自己忘记怎么配置，再又一次破坏环境之后，决定写一篇博客来记录一下。

创建python环境

conda create -n torch python=3.7
conda activate torch

在这一部分还是和原来的流程一样的，但值的注意的是spvnas代码中配置的pytorch环境需要和后续配置的torchsparse一致，因此需要配置对应的pytoch版本pytorch1.10.0和cuda11.3.

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch 

完成后检查

python -c "import torch; print(torch.__version__)"
>>> 1.10.0
python -c "import torch; print(torch.version.cuda)"
>>> 11.3

继续安装

conda install numba opencv
pip install torchpack
pip install --upgrade git+https://github.com/mit-han-lab/torchsparse.git

然后运行程序,如果出现报错

ModuleNotFoundError: No module named 'torch_scatter'
ModuleNotFoundError: No module named 'mpi4py'

安装对应的库

pip install mpi4py
pip install torch_scatter

结束完美运行
注意本人运行的程序为自改程序（为了获取mulran数据集的语义）