Pytorch编译安装warp-CTC


Pytorch 安装

进入Pytorch官网https://pytorch.org/,根据自己机器的环境选择相应的安装包,以conda+python3.6+cuda-8.0为例,

Pytorch编译安装warp-CTC_第1张图片

在终端输入:

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

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

Pytorch编译安装warp-CTC_第2张图片

你可能感兴趣的:(PyTorch)