PaddleX Local Installation Tutorial¶

❗Before installing PaddleX, please ensure you have a basic Python environment (Note: Currently supports Python 3.8 to Python 3.10, with more Python versions being adapted).

1. Quick Installation¶

Welcome to PaddleX, Baidu's low-code development tool for AI. Before we dive into the local installation process, please clarify your development needs and choose the appropriate installation mode.

PaddleX offers two installation modes: Wheel Package Installation and Plugin Installation. Below, we introduce their respective application scenarios:

1.1 Wheel Package Installation Mode¶

If your use case for PaddleX involves model inference and integration, we recommend the more convenient and lightweight Wheel package installation mode.

After installing PaddlePaddle (refer to the PaddlePaddle Local Installation Tutorial), you can quickly install the PaddleX Wheel package by executing the following commands:

❗ Note: Please ensure that PaddlePaddle is successfully installed before proceeding to the next step.

pip install https://paddle-model-ecology.bj.bcebos.com/paddlex/whl/paddlex-3.0.0b2-py3-none-any.whl

1.2 Plugin Installation Mode¶

If your use case for PaddleX involves custom development (e.g. retraining models, fine-tuning models, customizing model structures, customizing inference codes, etc.), we recommend the more powerful plugin installation mode.

After installing the PaddleX plugins you need, you can not only perform inference and integration with the supported models but also conduct advanced operations such as model training for custom development.

The plugins supported by PaddleX are listed below. Please determine the name(s) of the plugin(s) you need based on your development requirements:

👉 Plugin and Pipeline Correspondence (Click to Expand)

Pipeline	Module	Corresponding Plugin
General Image Classification	Image Classification	`PaddleClas`
General Object Detection	Object Detection	`PaddleDetection`
General Semantic Segmentation	Semantic Segmentation	`PaddleSeg`
General Instance Segmentation	Instance Segmentation	`PaddleDetection`
General OCR	Text Detection Text Recognition	`PaddleOCR`
Table Recognition	Layout Region Detection Table Structure Recognition Text Detection Text Recognition	`PaddleOCR` `PaddleDetection`
PP-ChatOCRv3-doc	Table Structure Recognition Layout Region Detection Text Detection Text Recognition Seal Text Detection Text Image Correction Document Image Orientation Classification	`PaddleOCR` `PaddleDetection` `PaddleClas`
Time Series Forecasting	Time Series Forecasting Module	`PaddleTS`
Time Series Anomaly Detection	Time Series Anomaly Detection Module	`PaddleTS`
Time Series Classification	Time Series Classification Module	`PaddleTS`
Image Multi-Label Classification	Image Multi-label Classification	`PaddleClas`
Small Object Detection	Small Object Detection	`PaddleDetection`
Image Anomaly Detection	Unsupervised Anomaly Detection	`PaddleSeg`

If the plugin you need to install is PaddleXXX, after installing PaddlePaddle (refer to the PaddlePaddle Local Installation Tutorial), you can quickly install the corresponding PaddleX plugin by executing the following commands:

git clone https://github.com/PaddlePaddle/PaddleX.git
cd PaddleX
pip install -e .
paddlex --install PaddleXXX

❗ Note: The two installation methods are not mutually exclusive, and you can install both simultaneously.

Next, we provide detailed installation tutorials for your reference. If you are using a Linux operating system, please refer to 2. Detailed Tutorial for Installing PaddleX on Linux.

2. Detailed Tutorial for Installing PaddleX on Linux¶

When installing PaddleX on Linux, we strongly recommend using the official PaddleX Docker image. Alternatively, you can use other custom installation methods.

When using the official Docker image, PaddlePaddle, PaddleX (including the wheel package and all plugins), and the corresponding CUDA environment are already pre-installed. You can simply obtain the Docker image and start the container to begin using it.

When using custom installation methods, you need to first install the PaddlePaddle framework, then obtain the PaddleX source code, and finally choose the PaddleX installation mode.

2.1 Get PaddleX based on Docker¶

Using the PaddleX official Docker image, create a container called 'paddlex' and map the current working directory to the '/paddle' directory inside the container by following the command.

If your Docker version >= 19.03, please use:

# For CPU
docker run --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-cpu /bin/bash

# For GPU
# For CUDA11.8
docker run --gpus all --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-gpu-cuda11.8-cudnn8.6-trt8.5 /bin/bash

# For CUDA12.3
docker run --gpus all --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-gpu-cuda12.3-cudnn9.0-trt8.6 /bin/bash

If your Docker version <= 19.03 and >= 17.06, please use:

Click Here

# For CPU
docker run --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-cpu /bin/bash

# For GPU
# For CUDA11.8
nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-gpu-cuda11.8-cudnn8.6-trt8.5 /bin/bash

# For CUDA12.3
nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8g --network=host -it ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.0.0b2-paddlepaddle3.0.0b2-gpu-cuda12.3-cudnn9.0-trt8.6 /bin/bash

If your Docker version <= 17.06, please update your Docker.
If you want to delve deeper into the principles or usage of Docker, please refer to the Docker Official Website or the Docker Official Tutorial.

2.2 Custom Installation of PaddleX¶

Before installation, please ensure you have completed the local installation of PaddlePaddle by referring to the PaddlePaddle Local Installation Tutorial.

2.2.1 Obtain PaddleX Source Code¶

Next, use the following command to obtain the latest PaddleX source code from GitHub:

git clone https://github.com/PaddlePaddle/PaddleX.git

If accessing GitHub is slow, you can download from Gitee instead, using the following command:

git clone https://gitee.com/paddlepaddle/PaddleX.git

2.2.2 Install PaddleX¶

After obtaining the latest PaddleX source code, you can choose between Wheel package installation mode or plugin installation mode.

If you choose Wheel package installation mode, execute the following commands:

cd PaddleX

# Install PaddleX whl
# -e: Install in editable mode, so changes to the current project's code will directly affect the installed PaddleX Wheel
pip install -e .

If you choose plugin installation mode and the plugin you need is named PaddleXXX (there can be multiple), execute the following commands:

cd PaddleX

# Install PaddleX whl
# -e: Install in editable mode, so changes to the current project's code will directly affect the installed PaddleX Wheel
pip install -e .

# Install PaddleX plugins
paddlex --install PaddleXXX

For example, if you need to install the PaddleOCR and PaddleClas plugins, execute the following commands to install the plugins:

# Install PaddleOCR and PaddleClas plugins
paddlex --install PaddleOCR PaddleClas

If you need to install all plugins, you do not need to specify the plugin names, just execute the following command:

# Install all PaddleX plugins
paddlex --install

The default clone source for plugins is github.com, but it also supports gitee.com as a clone source. You can specify the clone source using --platform.

For example, if you need to use gitee.com as the clone source to install all PaddleX plugins, just execute the following command:

# Install PaddleX plugins
paddlex --install --platform gitee.com

After installation, you will see the following prompt:

All packages are installed.

For PaddleX installation on more hardware environments, please refer to the PaddleX Multi-hardware Usage Guide