DCNv2安装适配pytorch各个版本
2023-12-14 19:14:04
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, 失败)
文章来源:https://blog.csdn.net/qq_36892712/article/details/134879119
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!