DCNv2安装适配pytorch各个版本

DCNv2安装适配pytorch各个版本

介绍

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

相关推荐
l12345sy2 天前
Day24_【深度学习—广播机制】
人工智能·pytorch·深度学习·广播机制
蒋星熠3 天前
如何在Anaconda中配置你的CUDA & Pytorch & cuNN环境(2025最新教程)
开发语言·人工智能·pytorch·python·深度学习·机器学习·ai
weiwei228443 天前
Torch核心数据结构Tensor(张量)
pytorch·tensor
wL魔法师3 天前
【LLM】大模型训练中的稳定性问题
人工智能·pytorch·深度学习·llm
技术小黑4 天前
Transformer系列 | Pytorch复现Transformer
pytorch·深度学习·transformer
DogDaoDao4 天前
神经网络稀疏化设计构架方法和原理深度解析
人工智能·pytorch·深度学习·神经网络·大模型·剪枝·网络稀疏
西猫雷婶4 天前
pytorch基本运算-Python控制流梯度运算
人工智能·pytorch·python·深度学习·神经网络·机器学习
ACEEE12224 天前
Stanford CS336 | Assignment 2 - FlashAttention-v2 Pytorch & Triotn实现
人工智能·pytorch·python·深度学习·机器学习·nlp·transformer
深耕AI5 天前
【PyTorch训练】准确率计算(代码片段拆解)
人工智能·pytorch·python
nuczzz5 天前
pytorch非线性回归
人工智能·pytorch·机器学习·ai