CenterPoint环境配置与安装
博主在配置CenterPoint的环境时,按照原链接安装时采了很多坑,总会各种报错,下面是成功运行的安装教程
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