ubuntu16.04安装mmdetection

26 篇文章 1 订阅
10 篇文章 0 订阅

拜读了B站的一位大神的视频,附上安装mmdetection视频,,若侵权会马上删除
mmdetection与mmcv版本对应关系链接

首先声明本人是ubuntu16.04 cuda10.1

第一步:需要创建虚拟环境open-mmlab,为啥创建不多说,不懂的先去学习一下,比较简单,就是创建的环境能保持独立,不会和其他环境混。
在终端执行:conda create -n open-mmlab
安装python版本:conda create -n python=3.7 -y
或者以上两步也可以总结成一步的命令:conda create -n open-mmlab python=3.7

第二步:激活虚拟环境,在终端执行:conda activate open-mmlab

第三步:装pytorch,建议使用conda安装,查看自己电脑cuda的版本,pytorch官网地址pytorch官当地址链接找对应的pytorch版本。我的电脑是cuda10.1
我的安装命令是:

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

后面换成了1.5版本

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch

第四步:安装mmcv,这一步很多人都会安装错误,直接conda install…这样及其不可取,因为这样很容易安装成最新版本,而导致跟自己电脑环境不匹配,所以推荐按照下列方法安装,后面链接是安装mmcv的索引,建议查看安装mmcv链接要求
之前我看到这里面有句:
在这里插入图片描述
Note: The pre-built packages provided above do not include all versions of mmcv-full, you can click on the corresponding links to see the supported versions. For example, you can click cu102-torch1.8.0 and you can see that cu102-torch1.8.0 only provides 1.3.0 and above versions of mmcv-full. In addition, We no longer provide mmcv-full pre-built packages compiled with PyTorch 1.3 & 1.4 since v1.3.17. You can find previous versions that compiled with PyTorch 1.3 & 1.4 here. The compatibility is still ensured in our CI, but we will discard the support of PyTorch 1.3 & 1.4 next year.
因此我之前一直安装pytorch1.4版本不成功,这句提示以后我换成了pytorch1.5

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{你的cuda版本}/{你的torch版本}/index.html

具体格式可以参照下面的版本,主要还是看你电脑的cuda和pytorch版本

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.4.0/index.html

第五步:安装mmdetection.

git clone https://github.com/open-mmlab/mmdetection.git或者git clone git://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e   或者python setup.py develop

测试环境是否可用

python tools/train.py -h

出现一下界面
在这里插入图片描述
在文件夹:./configs/gfl/gfl_r50_fpn_mstrain_2x_coco.py

中修改最大的训练epochs数
runner = dict(type=‘EpochBasedRunner’, max_epochs=100)
在这句代码中修改max_epochs中的数字大小,就能控制训练的epochs
最后代码跑起来了:
在这里插入图片描述

祝大家顺利。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

kelly1250230225

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值