Preparation

Catalogue

• 1. Prepara Environment

  • 1.1 Install PaddlePaddle
    • 1.1.1 By Docker
    • 1.1.2 By pip
    • 1.1.3 Check Installation
  • 1.2 Get PaddleClas
  • 1.3 Install Requirements

• 2. 快速创建PaddlePaddle, PaddleClas环境

• 1. Environment requirements

• 2.(Recommended) Prepare a docker environment

• 3. Install PaddlePaddle using pip

• 4. Verify installation

Docker is recomended to run Paddleclas, for more detailed information about docker and nvidia-docker, you can refer to the tutorial. If you do not want to use docker, you can skip section 1.1.1 (Recommended) Install PaddlePaddle by docker, and go into section 1.1.2 Install PaddlePaddle by pip.

1. Prepara Environment

1.1 Install PaddlePaddle

• python 3.x
• cuda >= 10.2 (necessary if paddlepaddle-gpu is used)
• cudnn >= 7.6.4 (necessary if paddlepaddle-gpu is used)
• nccl >= 2.1.2 (necessary distributed training/eval is used)
• gcc >= 8.2

Recomends:

• When CUDA version is 10.2, the driver version >= 440.33;
• For more CUDA versions and specific driver versions, please refer to link.

1.1.1 (Recommended) Install PaddlePaddle by Docker

• Switch to the working directory

cd /home/Projects

• Create docker container The following commands will create a docker container named ppcls and map the current working directory to the `/paddle' directory in the container.

# For GPU users
sudo nvidia-docker run --name ppcls -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:2.4.2-gpu-cuda10.2-cudnn7.6-trt7.0 /bin/bash

# For CPU users
sudo docker run --name ppcls -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:2.4.2 /bin/bash

Notices:

• The first time you use this docker image, it will be downloaded automatically. Please be patient;
• The above command will create a docker container named ppcls, and there is no need to run the command again when using the container again;
• The parameter --shm-size=8g will set the shared memory of the container to 8g. If conditions permit, it is recommended to set this parameter to a larger value, such as 64g;
• You can also access DockerHub to obtain the image adapted to your machine;
• Exit / Enter the docker container:
  • After entering the docker container, you can exit the current container by pressing Ctrl + P + Q without closing the container;
  • To re-enter the container, use the following command:

  sudo docker exec -it ppcls /bin/bash

1.1.2 Install PaddlePaddle by pip

If you want to use PaddlePaddle on GPU, you can use the following command to install PaddlePaddle.

pip install paddlepaddle-gpu --upgrade -i https://mirror.baidu.com/pypi/simple

If you want to use PaddlePaddle on CPU, you can use the following command to install PaddlePaddle.

pip install paddlepaddle --upgrade -i https://mirror.baidu.com/pypi/simple

Note:

• If you have already installed CPU version of PaddlePaddle and want to use GPU version now, you should uninstall CPU version of PaddlePaddle and then install GPU version to avoid package confusion.
• You can also compile PaddlePaddle from source code, please refer to PaddlePaddle Installation tutorial to more compilation options.

1.1.3 Check Installation

import paddle
paddle.utils.run_check()

Check PaddlePaddle version：

python -c "import paddle; print(paddle.__version__)"

Note:

• Make sure the compiled source code is later than PaddlePaddle2.0.
• Indicate WITH_DISTRIBUTE=ON when compiling, Please refer to Instruction for more details.
• When running in docker, in order to ensure that the container has enough shared memory for dataloader acceleration of Paddle, please set the parameter --shm-size=8g at creating a docker container, if conditions permit, you can set it to a larger value.

1.2 Get PaddleClas

Clone PaddleClas source code

git clone https://github.com/PaddlePaddle/PaddleClas.git -b develop

If it is too slow for you to download from github, you can download PaddleClas from gitee. The command is as follows.

git clone https://gitee.com/paddlepaddle/PaddleClas.git -b develop

1.3 Install Requirements

• [Recommended] Installing from PyPI:

pip install paddleclas

• Please build and install locally if you need to use the develop branch of PaddleClas to experience the latest functions, or need to redevelop based on PaddleClas. The command is as follows:

pip install -v -e .