拜读了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
祝大家顺利。
|