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