1. 复现报错解决
1.1 sh build_and_install.sh错误
复现PointRcnn时,安装 sh build_and_install.sh时出现Fatal error: cuda.h, No such file or directory,以下方法解决。
1.在~/.bashrc
中设置环境变量
export CUDA_HOME=/home/usr/local/cuda-10.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
2.查看makefile文件是否对cuda的地址是/usr/local/cuda-10.0 有的默认是/usr/local/cuda
ln -s /usr/local/cuda-10.0 /usr/local/cuda
3.如果以上情况都不是,并且locate cuda.h 能找到正确位置,在terminal中如下操作
sudo apt-get install nvidia-cuda-toolkit
4.重启reboot。
1.2 plane数据集问题?
将lib\datasets\kitti_rcnn_dataset.py
文件里的279~282行注释掉了,不让他调用数据增强函数apply_gt_aug_to_one_scene(...)
。plane数据集也是可选的。
1.3 GPU不够
减小减小batch_size值。
2. 复现过程
pointRCNN原理与复现、pointrcnn复现+可视化、
2.1 安装anaconda
2.2 环境搭建
- 创建虚拟环境/进入
conda create -n pointrcnn python=3.6
#官方要求版本3.6以上
conda activate pointrcnn
- 安装pytorch 1.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
打开python,并运行:import torch
载入该模块,未报错则成功。
检查:conda list
存在torch即安装完成