当前位置:   article > 正文

【OpenMMLab】mmcv-full安装、mmdet、MMRotate、torch、cuda等版本对应

【OpenMMLab】mmcv-full安装、mmdet、MMRotate、torch、cuda等版本对应

OpenMMLab部分项目部署、mmcv-full安装、mmdet、MMRotate、torch、cuda版本对应问题

前言:

5/16毕业设计答辩了,走过路过的同学帮忙给这篇文章(旋转框标注方法改进): link 点个赞让数据好看些嘛,因为这个改进的软件是真的帮助了部分同学的

考完研最近在写本科毕业论文,要运行OpenMMLab上的项目,但发现它们的框架升级之后变化是很多的,要是安装的稍微有点不对项目是完全不认包的,一开始我天真的以为改完api就能用了,然后到github上一顿搜一顿改,搞了两天,当然是运行不起来的。官网是有推荐版本下载的地方,但是它那里推荐的都是mmcv,不是项目要求的mmcv-full;

像这种情况,它们升级将那个BaseSampler包放到了mmegine上面,怎么改都是改不好的
然后我想着能不能花点钱解决问题,以下是我在鱼找的同学给的价格,当然那个320的肯定是来搞笑的。
那个80的我没看到,不然我当时肯定让他帮我配置环境了,还好没看到

版本对应关系,主要是迁就MMRotate的版本

版本对应github链接: link
这个是我目前截到的图片

cuda、Pytorch从头开始配置环境

假设你有相关配置基础了啊,不可能说从小白开始教怎么安装python的。

  1. 第一步:查看所用项目的MMRotate的版本
    不一定是这个项目,只要是要用到MMRotate的都行
    少截一张图,看目录结构应该能找到吧

  2. 准备安装Pytorch的前置步骤
    我是跟B站这个老师学习怎么安装cuda版本的pytorch的:link
    这个老师介绍了怎么安装指定版本,这一点很重要!!!
    当你跟着老师进行到选择pytorch版本之后,你先暂停视频,
    5.** 查找mmcv-full对应版本**
    回到下面这张图,我们就以0.3.4来详细说一下,其他版本的一样一样的。
    没有图片描述
    要求的mmcv-full是为1.5.3到1.8.0间的,打开链接:link
    不要直接下载,标黄的地方要自己找,cu116的意思是cuda的版本为11.6,torch1.13.0的意思是pytorch版本为1.13.X,X为任意可下载到的值。
    https://download.openmmlab.com/mmcv/dist/cu116/torch1.13.0/index.html
    一直滑到最低下才能看见mmcv-full,如果cuda和torch版本有符合MMRotate要求的版本则说明你选择的cuda版本和torch版本符合要求。选择一个和python版本相同的mmcv-full轮子文件下载。
    如果没有相应的mmcv-full或者网站压根打不开,那么则说明cuda版本不应该选择这个,建议是直接根据MMRotate版本,如百度 :安装mmcv-full 1.5.0,然后看看别人选择什么样的cuda和torch版本,不要自己呆呆的一个一个版本的试
    在这里插入图片描述

  3. 安装指定版本的pytorch
    根据上面选好的cuda和torch版本,结合B站老师的教程下载pytorch;

这个是下载mmcv-full 1.5.X的

pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html
  • 1

成功的标志

如果你在ide中打开项目没有,和mmcv、mmdet相关的包没有报错,就说明安装应该是每问题了。其他的一下像pandas就是正常安装即可。
要是安装有类似 from mmcv.runner import get_dist_info, init_dist 之中报错.runner找不到的,99%是版本报错
在这里插入图片描述

相关项目运行步骤

MMRotate运行步骤: link

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/607870
推荐阅读
相关标签
  

闽ICP备14008679号