CTC的安装 |
您所在的位置:网站首页 › 下载ctc › CTC的安装 |
CTC安装: 1. 在终端执行命令:git clone https://github.com/SeanNaren/warp-c) 如果使用的是pytorch0.4,执行终端命令; git clone --branch 0.4.1 https://github.com/SeanNaren/warp-ctc.git2. 打开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 安装完毕。 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |