【转载】THC/THC.h: No such file or directory报错解决

糖豆豆今天也要努力鸭

已于 2023-11-21 12:23:03 修改

阅读量4.7k

点赞数 41

分类专栏：杂谈文章标签： python ubuntu debug pytorch c++

于 2023-11-21 12:15:09 首次发布

原文链接：https://blog.youkuaiyun.com/qq_45934285/article/details/131120167

版权

杂谈专栏收录该内容

38 篇文章

订阅专栏

maskrcnn编译时报错，原因是我装的torch是1.13的，但THC.h文件在Pytorch1.10版之后被移除了。

解决方法：

将文件中的#include <THC/THC.h>注释掉
THCudaCheck函数报错。将所有该函数替换为AT_CUDA_CHECK
THCCeilDiv函数报错。在文件头部添加该函数：

template <typename T>
__host__ __device__ __forceinline__ T THCCeilDiv(T a, T b) {
  return (a + b - 1) / b;
}

THCudaMalloc THCudaFree THCState undefined
首先在用到Malloc和Free函数的文件中加入头文件

#include <ATen/cuda/ThrustAllocator.h>

注释掉THCState这句话，因为在新的pytorch中不需要使用该数据类型来生成Malloc所需空间。

THCState *state = at::globalContext().lazyInitCUDA(); // TODO replace with getTHCState

将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));

THCudaFree的替换方式如下：

// THCudaFree(state, mask_dev);
c10::cuda::CUDACachingAllocator::raw_delete(mask_dev);

另外补一些细枝末节的问题：
AttributeError: module ‘numpy‘ has no attribute ‘float‘
numpy版本高了，出现这个问题是因为np.float从1.24起被删除。所用的代码是依赖于旧版本的Numpy，降低numpy版本，降到1.23.5就行

----------------------------------------------一些碎碎念-----------------------------------------------
跟python setup.py build develop斗智斗勇已经有三个年头了。
在今天出现了全新的bug。
因为是4090 把torch从1.9换成了1.13的
结果就报错了^ _ ^

另外插一句 cudatoolkit11.3好像不能用在4090上 cuda.is_available()是false的
cudatoolkit11.7是可以的本机的CUDA也是11.7 其他版本的cudatoolkit还没试