pointpillar代码的开源地址:https://github.com/nutonomy/second
第一步:从github上拷贝代码
git clone https://github.com/nutonomy/second.pytorch.git
第二步:首先安装Anaconda创建虚拟环境,我的电脑Ubuntuan16.04安装的是anaconda3,cuda10的版本。和readme一样,使用conda install 安装虚拟环境中的相关配置
conda create -n pointpillars python=3.7 anaconda
source activate pointpillars
conda install shapely pybind11 protobuf scikit-image numba pillow
conda install pytorch torchvision -c pytorch
conda install google-sparsehash -c bioconda
到这里都不会出现问题,如果出现问题建议重装anaconda,或者直接重装系统。<如果是网络错误,就检查有没有conda换源>
注意运行完上面的命令后,虚拟环境中安装的是pytorch13.0.0。
第三步,用pip安装
pip install --upgrade pip
pip install fire tensorboardX
第三步结束后,开始安装 稀疏卷积部分,这里的bash build.sh命令必须在pytorch13.0 的环境下运行,否则会报错
git