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文件夹下,需要拷贝到上一层目录,要不然和安装目录对不上。

你可能感兴趣的:(深度学习)