OpenPCDet 环境安装

系统信息

Ubuntu 18.04
4 * 3090
128G memory
CUDA_VERSION 11.3

安装步骤

创建环境

conda create -n OpenPCDet python=3.8
先安装pytorch 
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

安装OpenPCDet 提供的requirements.txt 的内容

需要修改requirements.txt 修改过后的内容为

numpy
llvmlite
numba
#torch>=1.1
tensorboardX
easydict
pyyaml
scikit-image
tqdm
#torchvision
SharedArray
opencv-python
pyquaternion

执行安装openpcdet所需其它库

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

安装OpenPCDet

python setup.py develop

如上截图 是安装成功

---

额外安装

pip install kornia==0.5.8
pip install av2 -i https://pypi.tuna.tsinghua.edu.cn/simple

一定安装指定版本 kornia。未指定该版本可能会导致 安装kornia 的过程中重装pytorch