ubuntu22.04安装PaddleX3
PaddleOCR
安装过程可以参考PaddleX本地安装教程
我的电脑环境配置:
ubuntu22.04
cuda=11.8(之前安装的是12.4没有匹配的paddle-gpu;这里改成11.8)
一、安装基础环境
1、
conda create -n ppx1 python=3.10
2、
conda activate ppx1
3、
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=11.8 -c pytorch -c nvidia
4、
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
这里会在弹出来一个
A symlink already exists at /usr/local/cuda. Update to this installation? │ │ Yes │ │ No │ │
询问是否要更新现有的符号链接 /usr/local/cuda 到新安装的版本。这个选择取决于你的具体需求和当前环境配置
我这里选择yes(因为我之前装过其它版本;之前没装过就没有)
4、安装完成后需要配置环境变量:
sudo gedit ~/.bashrc
AI助手
在末尾添加:
export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.8/lib64
AI助手
5、最后;使更改生效:
source ~/.bashrc
二、下载安装PaddleX
我的是插件安装下载:
官网 PaddleX
1、下载安装包
git clone https://github.com/PaddlePaddle/PaddleX.git cd PaddleX
2、
cd PaddleX
3、# gpu,该命令仅适用于 CUDA 版本为 11.8 的机器环境
python -m pip install paddlepaddle-gpu==3.0.0b2 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
4、**安装PaddleX**
pip install paddlex==3.0.0b2
完成安装后会有如下提示,表示PaddleX安装成功!
All packages are installed.
三、模型推理
下载示例图片
新建一个py文件pp.py和output文件夹
touch pp.py
from paddlex import create_model
model_name = "PicoDet_layout_1x_infer"
model = create_model(model_name)
output = model.predict("layout.jpg", batch_size=1)
for res in output:
res.print(json_format=False)
res.save_to_img("./output/")
res.save_to_json("./output/res.json")
在这里面PaddleOCR下载一个推理模型解压
开始推理
python pp.py
完成
或者
以通用 OCR 产线为例:
paddlex --pipeline OCR --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_002.png --device gpu:0
(这个可以不用安装)
paddlex --install --platform gitee.com
参考:PaddleX3安装教程_paddlex安装-CSDN博客