second/pointpillars配制记录

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.csdn.net/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环境变量的问题。。。。下面是我修改好后的环境变量

second/pointpillars配制记录_第1张图片

6.最后训练的时候也报了一些错,大概是numpy版本的问题,把numpy降到1.17就没问题了。

7.现在numba还是有一些warning,但是我没管他,毕竟就只是一些warning哈哈哈哈哭了

三、效果图

second/pointpillars配制记录_第2张图片

我的邮箱是[email protected],有问题的小伙伴可以留言,尽力回复大家

 

你可能感兴趣的:(ubuntu,自动驾驶,深度学习)