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, 失败)

相关推荐
Vertira9 天前
PyTorch中的permute, transpose, view, reshape和flatten函数详解(已解决)
人工智能·pytorch·python
匿名的魔术师9 天前
实验问题记录:PyTorch Tensor 也会出现 a = b 赋值后,修改 a 会影响 b 的情况
人工智能·pytorch·python
Ven%9 天前
PyTorch 张量(Tensors)全面指南:从基础到实战
人工智能·pytorch·python
RockLiu@8059 天前
PyTorch 实现的 GlobalPMFSBlock_AP_Separate:嵌套注意力机制在多尺度特征聚合中的应用
人工智能·pytorch·python
苏苏susuus9 天前
深度学习:PyTorch人工神经网络优化方法分享(1)
人工智能·pytorch·深度学习
聚客AI9 天前
GPU利用率90%的秘诀:PyTorch/TensorFlow高阶优化技巧
人工智能·pytorch·tensorflow
杨靳言先9 天前
✨通义万相 2.1(Wan2.1)环境搭建指南:基于 CUDA 12.4 + Python 3.11 + PyTorch 2.5.1 GPU加速实战
人工智能·pytorch·python3.11
Ven%10 天前
矩阵阶数(线性代数) vs. 张量维度(深度学习):线性代数与深度学习的基石辨析,再也不会被矩阵阶数给混淆了
人工智能·pytorch·深度学习·线性代数·矩阵·tensor·张量
三道杠卷胡10 天前
【AI News | 20250623】每日AI进展
人工智能·pytorch·python·语言模型·github
IAM四十二10 天前
Pytorch风格迁移的Android实现
android·人工智能·pytorch