前置准备:Anaconda、编译安装CUDA10.1

地址:https://github.com/open-mmlab/OpenPCDet/blob/master/docs/INSTALL.md

1. 创建conda虚拟环境

conda create -n torch13 python=3.6
conda activate torch13

# conda install cudatoolkit=10.1
# conda install cudnn==7.6.5

2. 安装pytorch1.3.0

安装方法可参见之前的博客https://blog.csdn.net/hesongzefairy/article/details/108009749,可解决下载速度慢的问题

3.下载OpenPDect包

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

很慢的话可以翻墙自己去github下载

4. 安装其他依赖包

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

5. 安装spconv(重点部分)

地址:https://github.com/traveller59/spconv

目前spconv 1.2.1版本在pytorch1.4.0下配置存在一些问题,所以一开始就安装pytorch1.3.0

首先使用git下载spconv的源码repo,注意一定要使用--recursive

git clone https://github.com/traveller59/spconv --recursive

安装boost headers

sudo apt-get install libboost-all-dev

进入spconv的文件夹,生成spconv的wheel

cd spconv
python setup.py bdist_wheel

通过wheel安装spconv

cd dist
pip install spconv-1.2.1-cp36-cp36m-linux_x86_64.whl

6. 进入OpenPCDet文件夹安装pcdet

python setup.py develop

7. 验证

终端进入python,import pcdet无报错即安装完成

 

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐