您的足迹:首页 > Linux&Python >ubuntu18.04服务器使用GPU部署学习后的中文OCR识别教程

ubuntu18.04服务器使用GPU部署学习后的中文OCR识别教程

微信截图_20200629103920.png

准备工作:

1.本文使用github开源OCR支持GPU和CPU运算:https://github.com/myhub/tr

2.服务器环境ubuntu18.04+NVIDIA 1080 TI

GPU运算环境要求:

安装CUDA 10.1以及cuDNN 7.6.5

安装环境:

1.请先使用命令查看下驱动以及GPU的版本:

$ nvidia-smi

安装CUDA10.1需要更高版本的驱动,要求驱动版本表如下图:

微信截图_20200629142543.png

下载驱动地址:https://www.nvidia.cn/geforce/drivers/


2.驱动安装好以后,我们安装cuda10.1,如果可以访问官网下载最好,我这里因为访问不了官网,所以找了一个已经下好的cuda10.1版本的包

下载地址:https://quqi.gblhgk.com/s/1832596/UHGmJKvg7Z08WSMZ


3.安装cudnn7.6.5,因为官网无法提供下载原因,找了一个版本,现在提供下载

下载地址:https://quqi.gblhgk.com/s/1832596/UHGmJKvg7Z08WSMZ


4.全部安装完成后,按要求运行tr即可。

ps:还需要安装python环境


测试数据结果:

1.  8核8g cpu OCR识别效率每张A4纸识别结果平均需要3~5s时间,1200%以上的cpu使用率,50%以内的内存使用率

2.  GPU服务器计算A4纸识别结果平均需要0.5s时间,50%以下的gpu使用率,2000M内存使用






本博客所有文章如无特别注明均为原创。作者:cc复制或转载请以超链接形式注明转自 .NET程序员之路
原文地址《ubuntu18.04服务器使用GPU部署学习后的中文OCR识别教程

相关推荐

发表评论

路人甲 表情
Ctrl+Enter快速提交

网友评论(0)