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
如果输出下面的结果,表示安装正确,可以使用: