系统相关
首页 > 系统相关> > ubuntu16.04安装mmdetection

ubuntu16.04安装mmdetection

作者:互联网

拜读了B站的一位大神的视频,附上安装mmdetection视频,,若侵权会马上删除

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

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

第二步:激活虚拟环境,在终端执行: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

第四步:安装mmcv,这一步很多人都会安装错误,直接conda install…这样及其不可取,因为这样很容易安装成最新版本,而导致跟自己电脑环境不匹配,所以推荐按照下列方法安装,后面链接是安装mmcv的索引

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
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e   或者python setup.py develop

祝大家顺利。

标签:ubuntu16.04,mmcv,detection,pytorch,conda,install,版本,安装
来源: https://blog.csdn.net/kelly1250230225/article/details/122263009