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)
还没有任何评论哟~
