npu进行paddle框架适配
收藏
软件版本:
-- CANN 版本:5.1.Rc2
--Pytorch版本:1.8.1
--Python 版本:Python 3.7.5
--操作系统版本:Ubuntu 18.04
1、代码如下:
from paddleocr import PaddleOCR, draw_ocr
ocr = PaddleOCR(use_angle_cls=False,
show_log=False,
use_gpu=True,
det_model_dir='./inference/ch_ppocr_server_v2.0_det_infer/',
rec_model_dir='./inference/ch_ppocr_server_v2.0_rec_infer/',
cls_model_dir='./inference/ch_ppocr_server_v2.0_cls_infer/',
total_process_num=6,
use_mp=True)
2、安装paddleocr、paddle依赖后,依旧报错,根据百度结果提示,继续安装paddlepaddle,在npu环境下,paddlepaddle无法匹配,安装paddlepaddle-npu,但是安装后,运行会提示退回至CPU,不在npu运行
请问这种情况要怎么搞????
0
收藏
请登录后评论
可以把你的npu版本的paddle发我用一下吗
可以把你的npu的paddle发我一下吗