MMDetection框架入门教程之环境部署

创建虚拟环境

打开Anaconda Prompt,创建一个新的虚拟环境mmlab。注意我这里选择了python3.6,因为在python3.7下出现了mmdet3d.apis中的类无法无法import的情况(但mmdet3d可以import),暂时不清楚原因。

conda create -n mmlab python=3.6

然后切换到该环境下:

activate mmlab

安装pytorch

在这里插入图片描述
根据Pytorch官网的教程安装Pytorch,注意根据自己电脑的CUDA版本选择。在命令行运行nvcc -V查看当前cuda版本:
在这里插入图片描述
另外,当前(2021.10.23)MMCV最新仅支持到pytorch-1.9,安装最新的pytorch1.10会导致MMDetection无法使用,因此安装的时候需要指定pytorch版本为1.9。

conda install pytorch==1.9.1 torchvision torchaudio cudatoolkit=10.2 -c pytorch

这个时间可能会很漫长,耐心等待即可,安装完成后进行测试,看pytorch是否可用。
在这里插入图片描述

MMCV安装

MMCV有两mmcv-full和mmcv两个版本,两者差别在于是否包含CUDA操作,如果不需要使用CUDA可以安装mmcv,不过官方还是推荐安装完整版的mmcv-full

pip install mmcv-full

需要的依赖包很多,耐心等待下载。

在这里插入图片描述

MMDetection安装

在这里插入图片描述
我们需要使用的是mmdection,这里还有其他标准,如mmsegmentation等。
首先从github上下载mmdetection的源码,然后根据requirements.txt安装所需的依赖库,最后执行setup.py安装mmdetection。

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

将文件下载后打开,选择我们创建好的环境。
在这里插入图片描述

安装所需包:pip install -r requirements.txt

在这里插入图片描述

requirements内容,其分别去运行对应文件中要求的包。

-r requirements/build.txt
-r requirements/optional.txt
-r requirements/runtime.txt
-r requirements/tests.txt

这里运行到最后爆了个错误:Cannot find command ‘git’,没有 git 包,安装即可:

conda install git

在这里插入图片描述
完成后在pycharm的Terminal中再次执行pip install -r requirements.txt命令。再次报错:在这里插入图片描述
可能是由于pip版本太低导致的,按其要求更新即可,完成后再次执行。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

彭祥.

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

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

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

打赏作者

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

抵扣说明:

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

余额充值