second/pointpillars配置记录
一、环境配置
系统是ubuntu18.04
显卡是2080Ti
cuda版本为10.0
cudnn版本为7.3.1
python版本为3.7.3
pytorch版本为1.3
spconv版本为1.2(写这篇文章时的最新版,他需要python3.7和pytorch1.3以上)
Cmake版本为3.13.2
以上是环境版本配置,非常感谢https://blog.youkuaiyun.com/r1141207831/article/details/103756292 这篇博文的博主的详细步骤,给了我很大的帮助
二、安装步骤
https://github.com/traveller59/second.pytorch
官网的安装步骤比较详细,下面对我踩坑的点做出记录。
1.Cmake的版本必须在3.12.3以上,如果可以的话直接安装3.12.3即可(不建议卸载后再安装,可以直接覆盖)。
2.如果你是用的spconv最新版本,建议用python3.7及以上 pytorch1.3及以上。我用的python3.7和pytorch1.3没问题
3.torch 和 torchvision的版本要对应好,还有和你的cuda版本也要对应好
4.spconv代码克隆的时候要耐心点,中间出错了要重新来(多开几个终端一起下,总有一个会成功)
5.最恶心的一个点:我在运行
python create_data.py kitti_data_prep --root_path=data/KITTI_DATASET_ROOT
这条指令时出了一个关于cuda的错,弄了一天都没解决,最后发现是cuda环境变量的问题。。。。下面是我修改好后的环境变量
6.最后训练的时候也报了一些错,大概是numpy版本的问题,把numpy降到1.17就没问题了。
7.现在numba还是有一些warning,但是我没管他,毕竟就只是一些warning哈哈哈哈哭了
三、效果图
我的邮箱是2107802361@qq.com,有问题的小伙伴可以留言,尽力回复大家