Maskrcnn/Faster-rcnn安装过程中使用高版本torch

文章讲述了在安装Maskrcnn/Faster-rcnn时遇到THC.h相关错误,如缺失文件、未定义函数等,提供了包括替换头文件、调整内存管理以及修复torch._six模块等问题的详细解决方案。

本项目是作者在安装Maskrcnn/Faster-rcnn时使用高版本torch 遇到的THC.h的错误总结

问题:THC相关的包很多都在pytorch更新的过程中被抛弃了,导致很多.cu文件在安装依赖项时出现了不匹配的问题

fatal error: THC/THC.h: No such file or directory

在/maskrcnn_benchmark/csrc/cuda文件夹中的所有以.cu结尾的代码文件中删除了下述头文件:

#include <THC/THC.h>

并且把所有的:

THCudaCheck(cudaGetLastError());

替换成:

AT_CUDA_CHECK(cudaGetLastError());

“THCudaCheck“ is undefined

将所有文件中的:

THCudaCheck

替换成 

C10_CUDA_CHECK

"THCCeilDiv" is undefined

将所有文件中的:

THCCeilDiv

替换成:

std::min

例如:

dim3 grid(std::min(THCCeilDiv(output_size, 512L
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值