【OpenMMLab】mmcv-full安装、mmdet、MMRotate、torch、cuda等版本对应
OpenMMLab部分项目部署、mmcv-full安装、mmdet、MMRotate、torch、cuda版本对应问题
前言:
考完研最近在写本科毕业论文,要运行OpenMMLab上的项目,但发现它们的框架升级之后变化是很多的,要是安装的稍微有点不对项目是完全不认包的,一开始我天真的以为改完api就能用了,然后到github上一顿搜一顿改,搞了两天,当然是运行不起来的。官网是有推荐版本下载的地方,但是它那里推荐的都是mmcv,不是项目要求的mmcv-full;
然后我想着能不能花点钱解决问题,以下是我在鱼找的同学给的价格,当然那个320的肯定是来搞笑的。
版本对应关系,主要是迁就MMRotate的版本
版本对应github链接: link
cuda、Pytorch从头开始配置环境
假设你有相关配置基础了啊,不可能说从小白开始教怎么安装python的。
-
第一步:查看所用项目的MMRotate的版本
-
准备安装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版本,不要自己呆呆的一个一个版本的试
-
安装指定版本的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
成功的标志
如果你在ide中打开项目没有,和mmcv、mmdet相关的包没有报错,就说明安装应该是每问题了。其他的一下像pandas就是正常安装即可。
要是安装有类似 from mmcv.runner import get_dist_info, init_dist 之中报错.runner找不到的,99%是版本报错
相关项目运行步骤
MMRotate运行步骤: link
更多推荐
所有评论(0)