背景
PaddlePaddle这东西的适配性做得很差,运行环境对包版本经常很严格。我们今天要处理的问题是复现一份基于paddleocr的代码,要求支持cpu推理和gpu单卡推理。
情况摸排
- 根据项目其他依赖,选择使用3.8版本的python.
- 通过前置测试,确定代码依赖paddlepaddle=2.5.2, paddleocr=2.0.1,在此环境下cpu推理成功。因此,我们为GPU推理选择安装2.5.2版本的paddlepaddle-gpu.
注意,如果是从头开始开发,建议直接选取最新版本的paddle和paddleocr,然后在后续开发中保持版本不变。
- 运行环境是实体宿主机,cuda版本12.0,安装位置在
/usr/local/cuda/bin
,nvcc -V
显示正常。
安装方式
- 根据Paddle2.5官方文档,CUDA 工具包 12.0 配合 cuDNN v8.9.1. 这个链接同时提供了 CUDA12.0 包含 cuDNN 动态链接库的 PaddlePaddle安装方式。
pip install paddlepaddle-gpu