CenterPoint环境配置与安装

• Requirements
• Basic Installation
• Advanced Installation
• • nuScenes dev-kit
  • Cuda Extensions
  • spconv
  • APEX

https://github.com/tianweiy/CenterPoint/blob/master/docs/INSTALL.md

Requirements

• Linux
• Python 3.6+
• PyTorch 1.1 or higher
• CUDA 10.0 or higher
• CMake 3.13.2 or higher
• spconv
• APEX

注意：

• spconv应该是与cuda相匹配的特定版本
• 经验法则是，你的pytorch cuda版本必须与你的系统的cuda版本匹配，其他cuda扩展才能正常工作。

Basic Installation

conda create --name centerpoint python=3.7
conda activate centerpoint
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
git clone https://github.com/tianweiy/CenterPoint.git
cd CenterPoint
pip install -r requirements.txt# add CenterPoint to PYTHONPATH by adding the following line to ~/.bashrc (change the path accordingly)
export PYTHONPATH="${PYTHONPATH}:PATH_TO_CENTERPOINT"

Advanced Installation

nuScenes dev-kit

git clone https://github.com/tianweiy/nuscenes-devkit# add the following line to ~/.bashrc and reactivate bash (remember to change the PATH_TO_NUSCENES_DEVKIT value)
export PYTHONPATH="${PYTHONPATH}:PATH_TO_NUSCENES_DEVKIT/python-sdk"

Cuda Extensions

# set the cuda path(change the path to your own cuda location) 
export PATH=/usr/local/cuda-10.0/bin:$PATH
export CUDA_PATH=/usr/local/cuda-10.0
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
bash setup.sh 

spconv

sudo apt-get install libboost-all-dev
pip install spconv-cu113 # 根据自身cuda版本进行安装

APEX

git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install