报错-mmdet/cuda编译报错: fatal error: THC/THC.h: No such file or directory

目录

• 现象
• 原因
• 解决方法

现象

当执行 python setup.py develop ，有如下报错：

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

原因

1.11版本后，Pytorch中的THC/THC 命名空间已失效，apex 也已将其删除，但是其中的函数已移至 ATen 命名空间。

解决方法

THCudaCheck函数->AT_CUDA_CHECK

注释头文件

// #include <THC/THC.h>

// #include <THC/THCDeviceUtils.cuh>

重新编译即可