DCNv2安装适配pytorch各个版本

lqjun08272023-12-15 11:34

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。

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

bash 复制代码

# 克隆项目
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, 失败）