赞
踩
在安装的时候遇到一点坑,比如环境问题,安装完能跑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 //需要比你之前的低
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。