报错-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
// #include
重新编译即可

你可能感兴趣的:(ubuntu,gpu算力,深度学习)