Advertisement

win10和warp-CTC的pytorch绑定

阅读量:

warp-CTC安装

http://www.pianshen.com/article/862428080/

warpctc_pytorch 编译不成功的解决办法

warp-CTC属于百度开源的一个开放源代码的CTC代码库,并且可以在CPU和GPU上实现高效率地并行计算;该代码库实现了对CTC算法的高效并行处理。

warp-CTC安装:

$ git clone https://github.com/SeanNaren/warp-ctc.git
切换到仓库目录
创建build目录后切换到该目录
执行CMake构建
运行构建程序
移动到pytorch_binding父目录
运行Python安装命令

添加环境变量:

gedit ./.bashrc
export WARP_CTC_PATH=/home/xxx/warp-ctc/build

验证pytorch中warp-CTC是否可用GPU例子:

cd /home/xxx/warp-ctc/pytorch_binding/tests
python test_gpu.py
OK输出:

或:

导入PyTorch库
并加载必要的模块。
从warp-ctc-pytorch导入CTCLoss损失函数。
定义CTCLoss损失函数实例。
注释详细说明各部分的作用:
将概率张量转换为float类型,
其形状应满足seq

全部评论 (0)

还没有任何评论哟~