查看版本: conda list + 包名
版本:
cuda 10.1
python 3.7
torch 1.7.1
mmcv-full 1.4.0
mmdet 2.14.0
mmsegmentation 0.14.1
gcc 5.5
mmdet3d 0.17.2
安装cuda、cudnn
环境变量配置
export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-2019.1${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64/${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-10.1
安装mmdetection3d
安装torch
我用pip 安装的torch,更稳定些。
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
安装mmcv-full
注意适配自己的cuda和torch版本号
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.1/index.html
安装mmdet
pip install mmdet==2.14.0
安装mmseg
pip install mmsegmentation==0.14.1
如果此时编译mmdet3d可能编译不过去,gcc版本问题
gcc --version
或者 gcc -v
查看当前版本
7.5的版本网友说至少要配cuda11,我配的是cuda10,要降级gcc到5.5才可以,亲测7.5编译报错。
降级gcc
安装gcc5.5
sudo apt install gcc-5 g++-5
建立软连接
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 80 --slave /usr/bin/g++ g++ /usr/bin/g++-5
再次执行命令gcc -v
和 g++ -v
查看gcc、g++版本已是5.5
编译mmdet3d
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d/
pip install -v -e .
处理kitti数据集
python tools/create_data.py kitti --root-path ./data/kitti --out-dir ./data/kitti --extra-tag kitti
maybe出现这种错误!!!妖孽~~~
File "pycocotools/_mask.pyx", line 1, in init pycocotools._mask
ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 88 from C header, got 80 from PyObject
网友说pycocotools版本问题,原来是2.0.3,卸载并重新安装2.0.2
先躺一会儿