解决 /bin/bash^M: bad interpreter: 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>
重新编译即可