问题:import _ext as _backend ModuleNotFoundError: No module named '_ext'
最近在复现《Reference-based Image Super-Resolution with
Deformable Attention Transformer》这篇文章,在服务器上配置环境,配环境弄了好久,主要就是这个mmsr兼容性特别差,配了10天最后终于解决这个问题。
# 克隆项目
git clone https://github.com/lbin/DCNv2.git
cd DCNv2
# 显示远程所有分支
git branch -r
# 所有分支信息
# origin/HEAD -> origin/master
# origin/master
# origin/pytorch_1.11
# origin/pytorch_1.5
# origin/pytorch_1.6
# origin/pytorch_1.7
# origin/pytorch_1.9
# origin/pytorch_2.0.1
# 切换分支
git checkout pytorch_1.9
# 根据本地安装的pytorh版本切换到对应的分支,
# 比如当前安装版本为pytorch1.9是,切换到pytorch_1.9分支;若当前pytorch版本为2.0.1,则切换到pytorch2.0.1分支
# 安装DCNv2
python setup.py build develop
执行以上步骤,即可安装成功。同时也解决了一直编译c++失败的问题
我的环境是torch2.1.0+cuda12.1,服务器cuda为12.5,mmcv-full1.7.1