跑一个代码需要用到mmcv和mmseg
其中有两行代码:
from mmcv.runner import load_checkpoint
from mmseg.utils import get_root_logger
我先是按照官方推荐的安装方法去安装了mmcv和mmseg
pip install -U openmim
mim install mmcv
它会自动帮你安装mmengine
我的cuda是11.6 torch是2.0.1
他帮我安装的mmcv是2.1.0,mmengine是0.9.1
然后pip install mmsegmentation
安装的mmseg版本是1.2.1
安装完了之后出现问题:
mmcv.runner标红报错,说mmcv下面没有runner
ModuleNotFoundError: No module named 'mmcv.runner
我在网上搜索,有博客说不能安装mmcv,要安装mmcv-full
然后我把mmcv卸载,去安装mmcv-full(被坑惨了)
先是遇到一个问题,安装mmcv-full卡住
一直卡在buliding wheels for mmcv
我去网上一搜,很多遇到这个问题的
解决方法都是说,跟cuda版本不符,要安装对应的版本。
我试验了多个版本,都不行,安装不了
最后甚至在这个网站下载了离线的whl包。https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
最终安装成功,但是带来的问题更多,
找不到libcudaert.so.10.2
No module named ‘mmcv._ext‘
整整折腾了一个多小时,反正各种问题,大概都是在说,版本跟cuda、torch不匹配
(其实人家自动给你安装的mmcv就是符合你机器版本的,但是自动安装mmcv-full不了,手动安装的又不匹配,,,,)
最后换了思路去解决:
这两行代码的问题:
from mmcv.runner import load_checkpoint
from mmseg.utils import get_root_logger
首先:找不到mmcv.runner
我去看了源码,发现是版本升级的问题,我用的高版本的mmcv,结构已经变了
runner在mmengine下面,所以修改如下
from mmengine.runner import load_checkpoint
第二个错误:
mmseg.utils下面找不到get_root_logger
我感觉还是版本问题,高版本的结构变了
但是我去mmseg的源码下搜索了get_root_logger,搜不到,根本没有这个
然后我去mmcv和mmengine的源码下都搜索了,都没有这玩意
emm,然后看了下代码,这个东西就是用来记录日志的
from mmseg.utils import get_root_logger
logger = get_root_logger()
我看了下mmcv中日志是如何记录的
import logging
logger = logging.getLogger()
那既然这个找不到get_root_logger
直接把这句注释了
改成这样
import logging
logger = logging.getLogger()
运行发现没什么问题,解决了。
总结:
这个解决办法其实不好,跑别人的代码最好还是不用动别人的代码,配好相应的环境去运行,因为你动了某一处,不清楚会不会影响别的
但是我这是特殊情况,这个代码没有告知所需的库的版本,我也安装了各个版本的mmcv和mmcv-full最后实在是搞不定,才只好改了她的代码
另外,这玩意兼容性做的真的差,你升级版本你直接把原来的函数都给删除了,名字都改了,那之前的代码不都g了,让人怎么维护,怎么升级