mmdetection运行环境配置

纯小白一个,最近帮师兄跑一下代码,研究了好久,终于能跑mmdetection了,这篇就当做笔记了,免得换个电脑又不会了。mmdetection感觉坑很多,对版本要求很严格,所以安装时一定要仔细。

  1. 确认电脑的cuda版本。cmd里运行nvcc --version(一定要用这个方法,不要用网上从控制面板查看的方法)我的电脑是cuda10.2。
    在这里插入图片描述
  2. anaconda创建虚拟环境
 conda create -n mmlab python=3.7
  1. 进入虚拟环境
conda activate mmlab
  1. 下载pytorch和torchvision(下载地址 https://download.pytorch.org/whl/torch_stable.html),注意查看cuda版本(cmd里运行nvcc --version),这里使用pytorch1.6.0和torchvision0.7.0在这里插入图片描述
    在这里插入图片描述

  2. 下载完上面两个文件,进入这两个文件的文件夹, 通过pip安装这两个文件

  3. 修改numpy版本为1.17.1

pip install numpy==1.17.1
  1. 安装mmcv-full(full版本可以使用cuda,Windows下mmcv-full最新是1.15,注意修改地址里cuda和pytorch的版本)
pip install mmcv-full==1.1.5 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html 
  1. windows版mmcv只有1.1.5版本。所以要用2.7版的mmdetection。https://github.com/open-mmlab/mmdetection/archive/v2.7.0.zip 下载解压
  2. 进入mmdetection 文件夹
cd mmdetection
pip install -v -e .

这样基本就结束了,等后面运行程序的时候可能会缺少scipy,直接conda install scipy就好。
对于和我一样的小白来说需要注意的是怎么在anaconda prompt进入目标文件夹,基本操作是这样的,首先确认目标文件夹所在位置,比如说D:\mmdetection-2.7\demo,而你现在在C盘的文件夹下,那你需要先输入 D: 然后回车进入D盘,然后输入 cd mmdetection-2.7进入mmdetection-2.7文件夹,输入 cd demo进入demo文件夹就OK了。
所以在第5步和第9步进入文件夹的时候要进行上面的操作。

第一次写就当做一个笔记了,如果有人看的话,有错误求轻喷。
后面会写一下怎么把图片转换成coco格式,跑通mask-rcnn。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值