ubuntu22.04安装PaddleX3

发布于:2025-02-11 ⋅ 阅读:(70) ⋅ 点赞:(0)

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博客


网站公告

今日签到

点亮在社区的每一天
去签到