记录一下在安装CTC中遇到的问题

ctc安装:

1.遇到GPU版本高的问题:
打开makefile.txt,注释调35,36行的内容,添加适配你GPU的内容。其中参考https://github.com/baidu-research/warp-ctc/blob/master/CMakeLists.txt
在这里插入图片描述
此处我本机的cuda是11.2所以我就复制的cuda11.2部分,完美解决,版本过高问题。
在这里插入图片描述
2.接下来进行python setup.py install部分遇到新的问题,不知道为什么我的git checkout ac045b6072b9bc3454fb9f9f17674f0d59373789不能用。
在这里插入图片描述
所以不可避免的遇到了:

在这里插入图片描述
各种谷歌,各种百度之后,发现目前的CTC版本已经不断发展了,可以使用1.1以上的torch版本了(在这此前,上边的步骤都是使用的torch0.4),所以尝试着换个torch版本试试,本人换成了1.10,结果python setup.py install成功了。
在这里插入图片描述

3.cp libwarpctc.so …/…/usr/lib
成功 略过
4.开始测试是否安装成功:
cd /home/xxx/warp-ctc/pytorch_binding/tests
python test_gpu.py
问题:ModuleNotFoundError: No module named ‘pytest’
解决:conda install pytest
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值