second.pytorch安装踩坑实录

最新推荐文章于 2024-08-08 08:21:46 发布

阿瓜瓜瓜

最新推荐文章于 2024-08-08 08:21:46 发布

阅读量6.3k

点赞数 9

CC 4.0 BY-SA版权

分类专栏：深度学习环境配置

本文链接：https://blog.youkuaiyun.com/Left_control_right/article/details/109225993

本文详细记录了在安装和配置SECOND.pytorch (v1.5.1和v1.6) 以进行Kitti数据集3D目标检测过程中遇到的错误及解决方法，包括spconv版本匹配、模块导入错误、数据集处理、环境变量设置等。此外，还介绍了KittiViewerWeb的使用问题，如端口设置和代码修改。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写在前面

下载的时候一定要注意second.pytorch的版本，不然安装完spconv后，尝试Kitti Viewer Web的时候可能会一直报错“cannot import name 'rbbox_intersection'”，附上没有此问题的代码仓库：https://github.com/traveller59/second.pytorch/tree/v1.5.1

后续我又尝试了second.pytorch1.6版本，附上仓库链接：https://github.com/traveller59/second.pytorch,当我安装成功spconv1.2之后准备训练时，总是报错cannot import name 'non_max_suppression' from 'spconv.utils'，暂时没有解决办法。。后面又新建conda环境从新来过，pytorch1.3.1，torch0.4.2，开始并不行。报错libcudart.so.9.0: cannot open shared object file: No such file or directory，这里说torchvision版本不对，换成0.2.2，也不行。又pip install torchvision==0.4.2换回来。然后重新clone了仓库，改了改数据路径，居然能跑了。唉，跑吧，反正可以抓耗子了。。。

安装步骤

安装

克隆代码

###如果克隆失败，可以手动进入仓库网址下载，再上传服务器###
git clone https://github.com/traveller59/second.pytorch/tree/v1.5.1
cd ./second.pytorch/second


###这里是1.6版本，由于需要使用pointpillars，我又踩了一遍坑
git clone https://github.com/traveller59/second.pytorch
cd ./second.pytorch/second

安装依赖包（对1.6版本和1.5.1版本都是一样的）

###如果有Anaconda###
conda install scikit-image scipy numba pillow matplotlib
pip install fire tensorboardX protobuf opencv-python

###

最低0.47元/天解锁文章