当前位置:   article > 正文

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

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

解决方法:

  1. 将文件中的#include <THC/THC.h>注释掉
  2. THCudaCheck函数报错。将所有该函数替换为AT_CUDA_CHECK
  3. THCCeilDiv函数报错。在文件头部添加该函数:
template <typename T>
__host__ __device__ __forceinline__ T THCCeilDiv(T a, T b) {
  return (a + b - 1) / b;
}
  • 1
  • 2
  • 3
  • 4
  1. THCudaMalloc THCudaFree THCState undefined
    首先在用到Malloc和Free函数的文件中加入头文件
#include <ATen/cuda/ThrustAllocator.h>
  • 1

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

THCState *state = at::globalContext().lazyInitCUDA(); // TODO replace with getTHCState
  • 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));
  • 1
  • 2

THCudaFree的替换方式如下:

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

另外补一些细枝末节的问题:
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还没试

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/81899
推荐阅读
相关标签