0
点赞
收藏
分享

微信扫一扫

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

Java架构领域 2022-01-20 阅读 57

我的电脑环境

项目名称版本
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

这时候会报错
在这里插入图片描述
这里我通过排除法,对比一些旧的文件,发现是make.sh文件的问题,修改如下:

vim make.sh

修改如下
在这里插入图片描述
重新运行

./make.sh

在这里插入图片描述

成功!

举报

相关推荐

0 条评论