English | 简体中文

1 Environment Requirements

• PaddlePaddle (the version >= 2.3)
• OS: 64-bit
• Python 3(3.5.1+/3.6/3.7/3.8/3.9)，64-bit version
• pip/pip3(9.0.1+)，64-bit version
• CUDA >= 10.1
• cuDNN >= 7.6

2 Installation

2.1 Install PaddlePaddle

Please refer to the installation doc to install PaddlePaddle (the version >= 2.3).

Highly recommend you install the GPU version of PaddlePaddle, due to the large overhead of segmentation models, otherwise, it could be out of memory while running the models.

For example, run the following command to install Paddle with pip for Linux, CUDA 10.1.

python -m pip install paddlepaddle-gpu==2.3.2.post101 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

In Python interpreter, run the following command to confirm whether PaddlePaddle is installed successfully

>>> import paddle
>>> paddle.utils.run_check()

# If the following prompt appears on the command line, the PaddlePaddle installation is successful.
# PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.

# Confirm PaddlePaddle version
>>> print(paddle.__version__)

2.2 Install PaddleSeg

If you make modification to PaddleSeg/paddleseg, e.g, adding model and loss, you should install PaddleSeg from source.

If you only use PaddleSeg library, please install complied PaddleSeg.

2.2.1 Install PaddleSeg from Source

Clone the PaddleSeg repo from Github.

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

Run the following command, install PaddleSeg from source. If you make modification to PaddleSeg/paddleseg, it will be efficient without reinstallation.

cd PaddleSeg
pip install -r requirements.txt
pip install -v -e .

2.2.2 Install Compiled PaddleSeg

Run the following command, install the complied PaddleSeg.

pip install paddleseg

2.3 Verify Installation

In the root of PaddleSeg, run the following command. If there are no error in terminal log, you can use PaddleSeg to train, validate, test and export models with config method.

cd PaddleSeg
sh tests/install/check_predict.sh

3 Use PaddleSeg with Docker

Docker is an open-source tool to build, ship, and run distributed applications in an isolated environment. If you do not have a Docker environment, please refer to Docker. If you will use GPU version, you also need to install nvidia-docker.

We provide docker images containing the latest PaddleSeg code, and all environment and package dependencies are pre-installed. All you have to do is to pull and run the docker image. Then you can enjoy PaddleSeg without any extra steps.

Get these images and guidance in docker hub, including CPU, GPU, ROCm environment versions.

If you have some customized requirements about automatic building docker images, you can get it in github repo PaddlePaddle/PaddleCloud.