1. 程式人生 > >Pytorch編譯安裝warp-CTC

Pytorch編譯安裝warp-CTC


Pytorch 安裝

進入Pytorch官網https://pytorch.org/,根據自己機器的環境選擇相應的安裝包,以conda+python3.6+cuda-8.0為例,

在終端輸入:

conda install pytorch torchvision cuda80 -c pytorch

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="yourpath/warp-ctc/build"

驗證是否安裝好:

cd yourpath/warp-ctc/pytorch_binding/tests
python test_gpu.py

如果輸出下面的結果,表示安裝正確,可以使用: