我要评论
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, 失败）
发表评论