报错
目录
- 现象
- 原因
- 解决方法
现象
当执行 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>
重新编译即可