高版本pytorch和CUDA编译旋转目标检测DOTA_devkit包或mmdet或nms_rotated

最近被临时通知要参加个遥感目标检测竞赛,涉及的是旋转目标检测,试了几个方案,发现环境要求都是CUDA10.1或者10.2,pytorch版本也比较低,因为自己已经赔了高版本的CUDA11.7,也不想在装个低版本CUDA或者使用Docker环境,就强行在pytorch1.13和CUDA11.7环境中进行编译,将其中遇到的问题进行一下总结,给后续有相同需求的朋友一点参考,最后是可以编译成功的。

遇到的问题主要有以下几个:

解决方法:

1、将THC.h和THCDeviceUtils.cuh头文件注释掉

2、加入两个新的头文件ceil_div.h,ThrustAllocator.h

3、将代码中的THCCeilDiv()函数修改为at::ceil_div()

4、将代码中的THCState *state注释掉

5、将THCudaMalloc()和THCudaFree()修改为如下形式

6、将THCudaCheck()函数修改为AT_CUDA_CHECK()函数

7、然后重新编译,其他文件报错也是上面几个相同错误,采用相同修改方式即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值