Pytorch编译安装warp-CTC

版权声明：本文为博主原创文章，请尊重原创，转载请注明原文地址和作者信息！ https://blog.csdn.net/zzc15806/article/details/83999668

---

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

如果输出下面的结果，表示安装正确，可以使用：