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

我的电脑环境

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

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

安装流程

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

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

vim make.sh

修改如下
DCNv2在pytorch1.8.1的安装(经过测试)_第2张图片
重新运行

./make.sh

DCNv2在pytorch1.8.1的安装(经过测试)_第3张图片

成功!

你可能感兴趣的:(软件BUG解决,linux,深度学习,ubuntu,pytorch,深度学习)