今天試了一下baidu的torch ctc的安裝:
pip install torch-baidu-ctc
發現使用下面的代碼會報錯:
import torch
from torch_baidu_ctc import ctc_loss, CTCLoss
然後就源碼安裝吧:
git clone --recursive https://github.com/jpuigcerver/pytorch-baidu-ctc.git
cd pytorch-baidu-ctc
python setup.py build
python setup.py install
測試一下:
python -m unittest torch_baidu_ctc.test
這裏測試的時候注意,不要在編譯的pytorch-baidu-ctc裏面測試,否則會出現報錯哦。
No module named 'torch_baidu_ctc._C', ERROR when installing from source
參考文獻
[1]. No module named 'torch_baidu_ctc._C', ERROR when installing from source. https://github.com/jpuigcerver/pytorch-baidu-ctc/issues/1