DCNv2在pytorch1.8.1的安装(经过测试)

我的电脑环境

项目名称 版本
cuda版本 11.1
pytorch版本 1.8.1
ubuntu版本 18.04

在安装centerFusion(centerNet的多模态融合版本)时,遇到DCNv2不支持新版pytorch的问题,通过寻找其他博客,发现他们只找到了pytorch1.7的新版本,我测试适用于1.7的,也报了各种错,直到我找到了DCNv2某位大佬写的最新版,但是在安装的过程中,出现了很多问题,最后经过我的略微修改,可以运行了。

安装流程

#到你的路径上
git clone https://github.com/jinfagang/DCNv2_latest.git
cd ./DCNv2_latest
./make.sh

这时候会报错
DCNv2在pytorch1.8.1的安装(经过测试)
这里我通过排除法,对比一些旧的文件,发现是make.sh文件的问题,修改如下:

vim make.sh

修改如下
DCNv2在pytorch1.8.1的安装(经过测试)
重新运行

./make.sh

DCNv2在pytorch1.8.1的安装(经过测试)

成功!

上一篇:错误make: nvcc: Command not found 解决方案


下一篇:CentOS7.6下安装Redis5.0.7