mamba报错 No module named ‘basicsr’-模型训练论坛-模型训练-社区 | AheadAI
幻灯片-社区 | AheadAI
图标卡片
这是一个图标卡片示例
原创作品
这是一个图标卡片示例
灵感来源NEW
这是一个图标卡片示例
系统工具 GO
这是一个图标卡片示例

mamba报错 No module named ‘basicsr’

在运行MambaIR项目脚本的时候,尽管按照作者要求配置环境如下:

conda env update --name mambair --file environment.yaml

还是会报错找不到’basicsr’?

(mambair) root@autodl-container-18b640a1a0-4f72471d:~/autodl-tmp/MambaIR# python basicsr/test.py -opt options/test/test_MambaIR_SR_x4.yml
Traceback (most recent call last):
  File "basicsr/test.py", line 7, in <module>
    from basicsr.data import build_dataloader, build_dataset
ModuleNotFoundError: No module named 'basicsr'

这是是因为basicsr 作为本项目的一部分没有包含在 requirements.txtenvironment.yaml 中。但本地模块(包括 basicsr)并未被正确地注册到 Python 环境中,导致环境配置的过程中并没有导入这个模块导致找不到。也就是说,这里的项目工作以开发者模式引用了另一个名为basicsr的工作,因此需要先添加至python环境中,而不能简单的pip install解决。

解决办法:在test.py文件的第五行加入下面的代码,将路径改成项目文件夹的绝对路径

sys.path.append('/root/autodl-tmp/MambaIR')

即可解决,不需要手动安装。

请登录后发表评论

    没有回复内容