python3 torch-baidu-ctc安裝

今天試了一下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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章