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