tensorflow配置warp_ctc

本文详细介绍了如何在Ubuntu16.04环境下,使用TensorFlow1.8和GPU,安装并配置warp-ctc加速CTC函数。包括克隆项目、编译、设置环境变量、安装warpctc-tensorflow包的过程,以及解决libgomp.so.1版本问题的方案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

warp_ctc是百度的加速ctc函数,

warp-ctc: https://github.com/baidu-research/warp-ctc

wjc写的这篇博客参照:https://blog.youkuaiyun.com/qq_29133371/article/details/80407985

 他写比较多,而且安装的时候有点差异。

Anaconda 安装tensorflow参照官网即可。

环境:Ubuntu 16.04,tf 1.8, GPU

1:找一个位置,比如home目录,clone warp-ctc

git clone https://github.com/baidu-research/warp-ctc.git
cd warp-ctc

2:创建build目录(编译后文件所在目录)

mkdir build
cd build

3:编译

cmake ../ (../表示上级目录)
make

4:进入tensorflow_binding

cd tensorflow_binding


5:设置WARP_CTC_PATH(包含libwarpctc.so路径,编译后再build文件下可以找到),TENSORFLOW_SRC_PATH(tensorflow安装所在路径),CUDA_HOME(cuda根目录)(根据实际情况设置自己的path).在这里我用的是修改bashrc的方法。
比如我:

gedit  ~/.bashrc

然后添加环境变量:

export TENSORFLOW_SRC_PATH=/home/wujiacheng/anaconda3/lib/python3.6/site-packages/tensorflow(因为我tensorflow就装在这里)
export CUDA_HOME="/usr/local/cuda"
export WARP_CTC_PATH=$HOME/warp-ctc/build$WARP_CTC_PATH

----------------------------------------------------------------------------------------------------

我之前装cuda时是export CUDA_HOME=/usr/local/cuda-9.0

把这个改成export CUDA_HOME="/usr/local/cuda" 不改导致后面生成了一个名字叫cuda的文件,后面执行setup.py install出现了

File exiting 错误,把那个文件删了,在把CUDA_HOME设置好就好了

因为之前创建了软链接,这样不会有错的,然后再

source ~/.bashrc

6,在tensorflow_binding 目录

执行setup.py

python setup.py install

7,安装成功,conda list显示已经有了warpctc-tensorflow 0.2

但是

import warpctc_tensorflow

报错:libgomp.so.1: version `GOMP_4.0' not found

解决:把ubuntu系统中

/usr/lib/x86_64-linux-gnu/ 这个目录下的libgomp.so.1和libgomp.so.1.0.0

这两个文件(或者所有带libgomp.so这个名字的文件)复制到

/home/wjc/anaconda3/lib 这个目录下,选择替换

参照https://blog.youkuaiyun.com/dcrmg/article/details/78358279

https://blog.youkuaiyun.com/jiangpeng59/article/details/70213578

有些时候环境库或者环境变量配置好了,如果还没生效的话,就重新开个终端,

用了warpctc,感觉gpu的并没有提升多少,cpu的倒是快了蛮多

用gpu,warpctc和tensorflow自带的tf.nn.ctc_loss速度上感觉没什么差别

直接硬解lstm ctc问题,都很慢,还是得在lstm前面加cnn

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值