当执行 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 重新编译即可
京公网安备 11010502049817号