mmdetection3d系列--(1)安装步骤(无坑版)

  最近在看一些基于点云3d目标检测的文章,需要复现甚至修改一些算法,就找到了mmlab开源的mmdetection3d目标检测框架,方便后续学习。

    在安装的时候遇到一点坑,比如环境问题,安装完能跑demo但是不能跑训练测试问题等。在解决问题后还是完成了安装。在这里记录一下正确的安装流程,已备再次查阅,也给大家提供一点参考。

    首先建环境,最好是新建一个环境

conda create -n det python=3.8

#激活
conda activate det

    注意,以下步骤均在环境下进行!

    然后就是装pytorch,一定要根据cuda的版本来选择。用nvidia-smi查看可cuda版本, torch的网址:pytorch;   什么?还有没装cuda的?cuda的网站:cuda

  我的cuda版本是10.2,所以选择torch的安装命令的时候注意:

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch

     然后就正式进入mmdetection3d的依赖的安装环节了,首先下载工具mim:

pip install openmim

     然后一次下载5个东西,版本我都试好了,直接用:

#下载mmcv
mim install mmcv-full==1.6.0


#下载mmdet
mim install mmdet==2.27.0


#下载mmsegmentation
mim install mmsegmentation==0.30.0


#下载mmdet3d
mim install mmdet3d==1.0.0rc6


#顺便把可视化工具也下了,不然后面跑demo也省不掉

pip install open3d

    然后就可以下载源码加编译了:

git clone https://github.com/open-mmlab/mmdetection3d.git

cd mmdetection3d

pip install -e .

    到这里,就算完成了安装,可以简单验证一下,import不报错:

(det) xhao@zns:~$ python 
Python 3.8.16 (default, Jan 17 2023, 23:13:24) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import mmdet
>>> import mmdet3d
>>> exit()

后面有时间会讲验证,训练和测试。本人自己也在学习,一起进步哦~

-------------------------------------------------------------------

    在训练的时候有个问题:AttributeError:module ‘distutils‘ has no attribute ‘version

这个时候去环境里看一下setuptools的版本,问题可能出在setuptools的版本过高。直接安装即可:
 

pip install setuptools==59.5.0 //需要比你之前的低

AttributeError:module ‘distutils‘ has no attribute ‘version_looseversion = distutils.version.looseversion attr_Zizizirrr的博客-CSDN博客

  • 11
    点赞
  • 56
    收藏
    觉得还不错? 一键收藏
  • 9
    评论
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值