PaddleOCR 开源数据集icdar2015检测GPU上训练环境终于搭建成功!!!

在赤“巨资”买了一个二手的1080ti后终于确定入坑。(在大哥的建议下入手的这个是带水冷的。1080ti基本上没新的了,其他高端一点的也买不起)

在搭建检测训练环境时,各种尝试。刚开始在win10下搭了N次,默认的模型检测是没问题,但是每次都是一跑训练不一会就结束了,也没啥报错信息。不知是驱动跟cuda不匹配还是其他原因,不得而知。咨询群友,说是2.0版本在windows上不太友好,也有安装成功能跑起来的。多次尝试没有成功后,转战ubuntu。开始装的是ubuntu16.04,后来升级到了18.04。同样多次安装驱动,cuda后慢慢找到了“法门”。昨晚终于安装成功,而后立马安装paddlepaddel gpu环境,跑训练脚本。看看时间已经凌晨1:17了,再看看回显,嗯,这次应该是ok了(兴奋了那么0.01s)。然后听着散热器的风扇呼呼转动的声音,慢慢睡去。

原计划早上起来收结果,从回显看到早上八点半左右,大概还只迭代了一半(500张图)。那就只好晚上回来再看结果好了。

训练过程截图

补充:回显里时间不对,是因为时区没同步

哈哈,终于上车入门(keng)了。后面还有收集数据,标定,训练检测,训练识别,模型优化以及移动端部署,优化等事情要做,只有吭哧吭哧一点点搞了。



----经历近15个小时跑完检测训练----



PaddleOCR 开源数据集icdar2015检测GPU上训练环境终于搭建成功!!!_第1张图片





PS:上周三天假基本上都在装环境了

你可能感兴趣的:(PaddleOCR 开源数据集icdar2015检测GPU上训练环境终于搭建成功!!!)