Ubuntu + pytorch1.7 + rtx3090/3080 安装DCNv2 踩坑记录

由于RTX3090/3080只能安装cuda11,cuda11适配的pytorch只有1.7。所以DCNv2的安装包目前还没有适配的,由于刚出较新,网上的资料相对缺乏,经过一系列踩坑,终于找到一个解决方案。

我的配置:
RTX3090
Ubuntu 18.04
cuda11.1
pytorch1.7

往上的解决方案有:
日常填坑-DCNv2+pytorch1.7+vs2017编译
我试了试该方案,报错,帖子下面有我的评论,总之尝试未果。

在原版的DCNv2的讨论区有人给出了解决方案:
Pytorch 1.6-1.8 compatability - CUDA11/3090 ready #92

该作者的新工程地址:DCNv2 pytorch1.7
下载下来,直接编译可以通过,我的运行良好。

需要注意的是,作者把dcn-v2.py的文件放到了DCN文件夹下,需要拷贝到上一层目录,要不然和安装目录对不上。

  • 7
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 16
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值