当前位置: 首页 > 工具软件 > WARP-CTC > 使用案例 >

Ubuntu环境 warp-CTC安装

叶允晨
2023-12-01

warp-CTC安装

git clone https://github.com/SeanNaren/warp-ctc.git
cd warp-ctc
mkdir build; cd build
cmake ..
make
cd ../pytorch_binding
python setup.py install

添加环境变量

gedit ./.bashrc
export WARP_CTC_PATH=/workspace/xxx/warp-ctc/build
cp lib.linux-x86_64-3.5/warpctc_pytorch/_warp_ctc.cpython-35m-x86_64-linux-gnu.so /usr/lib/x86_64-linux-gnu/
cp warp-ctc/build/libwarpctc.so /usr/lib/x86_64-linux-gnu/

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

cd /home/xxx/warp-ctc/pytorch_binding/tests
python test_gpu.py
 类似资料: