Skip to content

Latest commit

 

History

History
63 lines (39 loc) · 1.75 KB

install_en.md

File metadata and controls

63 lines (39 loc) · 1.75 KB
Raw

Installation

Introducation

This document introduces how to install PaddleClas and its requirements.

Install PaddlePaddle

Python 3.6, CUDA 9.0, CUDNN7.6.4 nccl2.1.2 and later version are required at first, For now, PaddleClas only support training on the GPU device. Please follow the instructions in the Installation if the PaddlePaddle on the device is lower than v1.7

Install PaddlePaddle

pip install paddlepaddle-gpu --upgrade

or compile from source code, please refer to Installation.

Verify Installation

import paddle
paddle.utils.run_check()

Check PaddlePaddle version：

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

Note:

  • Make sure the compiled version is later than PaddlePaddle2.0rc.
  • 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 data read 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.

Install PaddleClas

**Clone PaddleClas: **

cd path_to_clone_PaddleClas
git clone https://github.com/PaddlePaddle/PaddleClas.git

Install requirements

pip install --upgrade -r requirements.txt

If the install process of visualdl failed, you can try the following commands.

pip3 install --upgrade visualdl==2.0.0b3 -i https://mirror.baidu.com/pypi/simple

What's more, visualdl is just supported in python3, so python3 is needed if you want to use visualdl.