THC/THC.h: No such file or directory 问题原因及一系列解决方案

问题原因

出现THC/THC.h: No such file or directory 这个问题的原因主要是:pytorch在1.11版本之后,THC/THC.h就被移除了,所以出现这个问题的原因是因为pytorch版本太高的原因。所以,解决这个问题一个最直观的方法就是降版本,但是不想将版本,可以按照下面的方法更改。

解决方案

将#include <THC/THC.h>相关的头文件都注释掉。然后加入下面三个头文件:

#include<ATen/ceil_div.h>
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>

这时候在编译是,会出现一堆方法函数找不到的情况,可以按照下面的替换方法一步一步更改:

1. identifier "THCudaCheck" is undefined: THCudaCheck 替换成 C10_CUDA_CHECK

2. error: identifier "THCCeilDiv" is undefined: 
这个问题有两种解决方案:
第一种:THCCeilDiv(x,y)的地方,用 (x+y-1)/y的形式替换
第二种:THCCeilDiv(x,y)用at::ceil_div(x,y)替换

3. identifier "AT_CHECK" is undefined: AT_CHECK替换为TORCH_CHECK

4. identifier THCudaMalloc、THCudaFree THCState undefined这三个函数未定义:
	4.1 THCudaMalloc: mask_dev = (unsigned long long*) THCudaMalloc(state, boxes_num * col_blocks * sizeof(unsigned long long));用下面的语句代替:
mask_dev = (unsigned long long*) c10::cuda::CUDACachingAllocator::raw_alloc(boxes_num * col_blocks * sizeof(unsigned long long));

	4.2 THCudaFree:THCudaFree(state, mask_dev) ——————> c10::cuda::CUDACachingAllocator::raw_delete(mask_dev);
	4.3 下面这句可以直接注释,不再需要:
	#THCState *state = at::globalContext().lazyInitCUDA(); 
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值