欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【探坑之旅】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:

【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss

4、 在build路径下,使用VS打开后缀为sln的文件:

【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss

 

 

【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss

 

然后点击“生成”-->"重新生成解决方案",完成。

 

5、进入pytorch_bindling文件,运行python setup.py install

如果是pytorch 1.x会报错:

【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss

如何解决???? 只能安装新的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

【探坑之旅】windows下安装CTC from warpctc_pytorch import CTCLoss

又出现了其它错误,。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

 

相关标签: 安装教程