码迷,mamicode.com
首页 > 其他好文 > 详细

CTC安装机器错误解决办法:binding.cpp:92:49: error: cannot convert ‘THCudaTensor*’ to ‘const THFloatTensor*’ for argument ‘1’ to ‘int64_t THFloatTensor_size(const THFloatTensor*, int)’

时间:2019-09-03 16:19:52      阅读:238      评论:0      收藏:0      [点我收藏+]

标签:title   执行   error:   tensor   解决   装机   tor   htm   code   

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

安装完毕。

CTC安装机器错误解决办法:binding.cpp:92:49: error: cannot convert ‘THCudaTensor*’ to ‘const THFloatTensor*’ for argument ‘1’ to ‘int64_t THFloatTensor_size(const THFloatTensor*, int)’

标签:title   执行   error:   tensor   解决   装机   tor   htm   code   

原文地址:https://www.cnblogs.com/elitphil/p/11453447.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!