问题原因
出现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_CHECK2. 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_CHECK4. 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();