Ubuntu18.04安装OpenPCDet:spconv一键安装

文章目录

  • 前言
  • 本机安装环境
  • 安装Spconv
  • 安装pcdet
    • clone 项目
    • 创建并配置虚拟环境
    • 安装pcdet
  • 总结


前言

最近在摸索点云数据的处理,希望实现对点云数据的目标检测。但是在复现PointPillars算法时遇到很多坑,特此记录一下。不建议在windows环境下测试程序,笔者自己测试下,windows总会遇到奇奇怪怪的报错。


本机安装环境

Unbuntu18.04
Python 3.7
Pytorch 1.10
cuda 11.1
spconv v2.1.22


安装Spconv

进入Spconv官网,根据自己的CUDA版本找到对应的install命令,直接输入即可。
Ubuntu18.04安装OpenPCDet:spconv一键安装_第1张图片

本次安装选用v2.1.22,安装命令为:pip install spconv-cu111

出现上图说明安装成功。


安装pcdet

clone 项目

git clone https://github.com/open-mmlab/OpenPCDet.git

创建并配置虚拟环境

创建

conda create -n OpenPCDet python==3.7

激活

conda activate OpenPCDet

安装环境

先单独安装torch
torch安装连接为:https://pytorch.org/get-started/previous-versions/
本次安装选用的命令行为:
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

torch安装完成后,cd 到clone好的OpenPCDet目录下:

pip install -r requirements.txt

安装pcdet

在OpenPCDet目录下:

python setup.py develop 

Ubuntu18.04安装OpenPCDet:spconv一键安装_第2张图片
出现上图说明安装成功。


总结

本次环境安装前前后后搞了两天,但最后还是跑出了点云的目标检测算法的demo,结果还算是好的。
Ubuntu18.04安装OpenPCDet:spconv一键安装_第3张图片

参考文档:
https://blog.csdn.net/weixin_48994268/article/details/116746453
https://blog.csdn.net/ylwhxht/article/details/121956936
如果阅读本文对你有用,欢迎一键三连呀!!!
2022年7月21日16:30:41
Ubuntu18.04安装OpenPCDet:spconv一键安装_第4张图片

你可能感兴趣的:(点云,深度学习,python,计算机视觉)