CTC安装及其错误解决办法:binding.cpp:92:49: error: cannot convert ‘THCudaTensor*’ to ‘const THFloatTensor*’ for argument ‘1’ to ‘int64_t THFloatTensor_size(const THFloatTensor*, int)’ binding.cpp:6:29: fatal error: torch/extension.h: No such file or directory

CTC安装:

1. 在终端执行命令:git clone https://github.com/SeanNaren/warp-c)

(效果如下图,大家不用管我前面括号的内容,那是我打开的虚拟环境)

2. 打开warp-ctc文件夹:cd warp-ctc

3.执行命令:git checkout ac045b6072b9bc3454fb9f9f17674f0d59373789

  (这条命令要执行,不然会出现binding.cpp:6:29: fatal error: torch/extension.h: No such file or directory这个错误.

  详情请见:https://www.cnblogs.com/elitphil/p/11451494.html)

4. 执行命令:mkdir build

5. 执行:cd build

6.执行:cmake ..

7. 执行:make

8.执行:cd ../pytorch_binding/

9.执行:python setup.py install

如果你出现以下错误:

 莫慌。按照如下解决方案进行pytorch_bingding/src/binding.cpp的代码段修改。这是造成上述错误的主要原因(这是我从别的博客学的,在这里主要作为自己学习的记录以及后人避坑)。

1. binding.cpp文件的92行修改成如下:
int probs_size = THCudaTensor_size(state, probs, 2);


2. binding.cpp文件的105行修改成如下:
void* gpu_workspace;
THCudaMalloc(state, &gpu_workspace, gpu_size_bytes);

解决。

10. 接下重新执行:python setup.py install

安装完毕。

猜你喜欢

转载自www.cnblogs.com/elitphil/p/11453453.html