mmdetection3d 环境配置

查看版本: 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

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

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 -vg++ -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

在这里插入图片描述
先躺一会儿

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值