paddleocr ubuntu16.04 cuda10.2 安装

环境安装(paddle版本2.6,cuda10.2 cudnn7)

默认已经下载完项目
1.创建新的conda虚拟环境

conda create -p /data/anaconda/envs/paddle python=3.7

2.激活进入

conda activate paddle

3.安装paddle-gpu

python -m pip install paddlepaddle-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple 
pip install "paddleocr>=2.0.1"

如果出现 protobuf 版本不符需改变版本并且更正 onnx版本 (降低)
protobuf版本为3.20.0 onnx版本更正为1.12.0

pip install "protobuf==3.20.0"
pip install "onnx==1.12.0"

运行

下载官网模型,有三个,一个是检测模型,第二个是方向模型,第三个是文字检测模型,项目中新建目录inference,并将下载好的模型解压放在目录下。

python tools/infer/predict_system.py --image_dir ./img_dir/2009291F12H434-0-lp.jpg --det_model_dir ./inference/ch_PP-OCRv3_det_infer/ --rec_model_dir ./inference/ch_PP-OCRv3_rec_infer/ --cls_model_dir ./inference/ch_ppocr_mobile_v2.0_cls_infer/
在这里插入图片描述
测试照片为上图
运行时cpu正常,结果为
在这里插入图片描述
gpu报错,在这里插入图片描述

仔细查看原因 cuda10.2系统自带cudnn是8. x版本,但是paddle对应cuda10.2为7.x版本。
解决方案1:将usr/local/cuda-10.2/include中的cudnn替换为7.x 由于是公司服务器换了大家环境不能用了
解决方案2:在conda 虚拟环境中下载10.2和cudnn7.x 这时需要将conda环境加入系统环境变量否则会报错
FatalError: Segmentation fault is detected by the operating system.
使用conda list 查看
在这里插入图片描述

加入虚拟环境命令行:export LD_LIBRARY_PATH=/***/anaconda3/envs/padocr/lib:$PATH 你自己的conda环境,具体位置可以用conda env list 查看
gpu成功运行结果为
在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值