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

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

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

三、效果图

我的邮箱是2107802361@qq.com,有问题的小伙伴可以留言,尽力回复大家

 

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值