DCNv2安装适配pytorch各个版本

介绍

原始项目:https://github.com/CharlesShang/DCNv2
注意原始项目只有低版本的pytorch安装

解决方案

使用Fork版本(推荐)☆☆☆☆☆

根据https://github.com/CharlesShang/DCNv2/issues/74,推荐使用别人fork后处理的一个项目,项目包含支持对pytorch多个版本的处理,项目地址https://github.com/lbin/DCNv2

克隆项目,切换分支,执行安装。详细步骤如下:

# 克隆项目
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

执行以上步骤,即可安装成功。

DCNv2_latest项目(不推荐)

对于DCNv2_latest,支持的pytorch版本并不全面,不推荐使用。测试了一下,结果如下:
master分支支持pytorch1.11+(测试pytorch2.0.1+cuda118, 成功)
pytorch1.6分支支持pytroch之前版本(测试pytorch1.10.0+cuda111, 失败)

  • 11
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
如果出现"No module named 'dcn_v2'"的错误,可能是因为缺少模块或者模块的导入路径不正确。解决这个问题可以尝试以下几个步骤: 1. 首先,请确保你的pytorch版本是正确的,并且已经安装了相应的依赖。可以使用命令`pip show torch`来检查你当前使用的pytorch版本。如果版本不正确,你可以尝试升级或降级到兼容的版本。 2. 检查你的C语言编译是否正确。在使用某些库时,需要先进行C语言的编译,否则会报错。你可以参考相关文档或者官方指南来进行正确的编译步骤。 3. 确保你的模块导入路径正确。有时候,模块的导入路径可能不正确,导致找不到相应的模块。你可以尝试将导入路径修改为正确的路径。比如,将"No module named 'dcn_v2'"修改为"from model.defconv.functions.deform_conv_func import DeformConvFunction"。 4. 如果你使用的是python3,可能是因为缺少相关依赖文件。请检查你的python3安装目录下是否缺少必要的依赖文件。比如,你可以将解压文件放在python安装目录/lib/python3/lib-dynload目录中。 通过以上几个步骤,你应该能够解决"No module named 'dcn_v2'"的问题。如果还有其他问题,请提供更多的信息,我将尽力帮助你解决。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [2020-09-08](https://blog.csdn.net/Douzaikongcheng/article/details/108472038)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [解决centos python 3 ModuleNotFoundError: No module named '_bz2'](https://download.csdn.net/download/sdyyang/12565810)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值