ubuntu22.04安装PaddleX3

news2024/12/24 13:28:50

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下载一个推理模型解压

开始推理