【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss
程序员文章站
2022-06-06 10:32:31
...
下面是一次失败的惨痛经历,为pytorch0.4.0编译warp-ctc。到了下面第六步,我才指导pytorch 1.1以上的版本已经将ctc收编了,糟心,搞了一下午。
# 根本不需要再编译ctc了
from torch import nn
criterion = torch.nn.CTCLoss()
想了想,还是记录这次探坑之旅,torch 1.0版本就不用编译ctc啦。
1、下载wrap-ctc: https://github.com/amberblade/warp-ctc
2、安装cmake, 自行百度
3、打开cmake,配置两个路径,点击Configure和Generate:
4、 在build路径下,使用VS打开后缀为sln的文件:
然后点击“生成”-->"重新生成解决方案",完成。
5、进入pytorch_bindling文件,运行python setup.py install
如果是pytorch 1.x会报错:
如何解决???? 只能安装新的0.4.1版本的环境
conda create --name pytorch_gpu040 python=3.7
activate pytorch_gpu040
# 使用清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch=0.4.1 cuda100
cuda100代表CUDA版本(https://pytorch.org/get-started/previous-versions/)
pytorch 1.x安装:https://blog.csdn.net/kan2281123066/article/details/93789303
6、执行python setup.py install
又出现了其它错误,。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。