DCNv2安装适配pytorch各个版本

DCNv2安装适配pytorch各个版本

  • 介绍
  • 解决方案
    • 使用Fork版本(推荐)☆☆☆☆☆
    • DCNv2_latest项目(不推荐)

介绍

原始项目: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, 失败)

你可能感兴趣的:(Pytorch,pytorch)