paddleocr预测时gpu没有使用

[yuzhobotianzhiyu]

我分别用cpu和gpu预测5张图片，两者速度都是一样。在ai studio服务器gpu环境中运行的，安装了paddlepaddle和paddlepaddle-gpu两个环境，想问下如何使用use_gpu开启加速。目前不论是服务器还是自己本地运行，开不开gpu速度都是一样的

[ShenYuhan]

用的哪个版本的paddlehub，用的哪个版本的ocr模型

[yuzhobotianzhiyu]

用的哪个版本的paddlehub，用的哪个版本的ocr模型

2.0版本的paddlehub，chinese_ocr_db_crnn_mobile 8.4M的模型

[ShenYuhan]

确认下是不是最新的1.0.4版本？我这边测试一下

[yuzhobotianzhiyu]

确认下是不是最新的1.0.4版本？我这边测试一下

是最新版，这是源码，use_gpu设置True和False速度一样： import paddlehub as hub

加载移动端预训练模型

ocr = hub.Module(name="chinese_ocr_db_crnn_mobile") import matplotlib.pyplot as plt import matplotlib.image as mpimg import time import cv2 import os os.environ['CUDA_VISIBLE_DEVICES']='0' results = ocr.recognize_text(paths=['6.jpg'], use_gpu=False, # 是否使用 GPU；若使用GPU，请先设置CUDA_VISIBLE_DEVICES环境变量 output_dir='ocr_result', # 图片的保存路径，默认设为 ocr_result； visualization=False, # 是否将识别结果保存为图片文件； box_thresh=0.5, # 检测文本框置信度的阈值； text_thresh=0.5) # 识别中文文本置信度的阈值；