paddleocr 2.7.3 GPU和CPU推理结果不一致 about paddleocr HOT 7 CLOSED

旧的显卡paddle不支持

我的显卡是H800，然后paddlepaddle没有装CPU版本的，只存在GPU版本的paddlepaddle，重新试了下cuda 11.8，依然还是识别为none。重新看了下paddle那边的issue，貌似是因为paddlepaddle暂时不支持H系列显卡。

from paddleocr.

换个gpu试试

from paddleocr.

import cv2
from paddleocr import PaddleOCR
import paddle
# 使用第一块GPU
paddle.set_device('gpu:0')

img_path = "general_ocr_008.png"
frame = cv2.imread(img_path)

ocr = PaddleOCR(use_angle_cls=True, lang='ch', show_log=False, use_gpu=True)
result2 = ocr.ocr(frame, cls=False)
print(result2)
# result2 和result结果不同

假如是多卡，建议paddle.set_device('gpu:0')设置一下用哪张卡

from paddleocr.

import cv2
from paddleocr import PaddleOCR
import paddle
# 使用第一块GPU
paddle.set_device('gpu:0')

img_path = "general_ocr_008.png"
frame = cv2.imread(img_path)

ocr = PaddleOCR(use_angle_cls=True, lang='ch', show_log=False, use_gpu=True)
result2 = ocr.ocr(frame, cls=False)
print(result2)
# result2 和result结果不同

假如是多卡，建议paddle.set_device('gpu:0')设置一下用哪张卡

刚刚尝试了一下，还是一样，8张卡返回的结果都是None。这个检测识别结果会和cuda版本有关系吗？我的cuda版本是12.2，paddlepaddle我看官方文档是只写了支持cuda 12.0的。 @jingsongliujing @GreatV

from paddleocr.

from paddleocr.

旧的显卡paddle不支持

from paddleocr.

可以尝试自己编译一个paddle框架

from paddleocr.