@@ -8,7 +8,7 @@ This document introduces how to install PaddleClas and its requirements.
## Install PaddlePaddle
Python 3.x, CUDA 10.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](http://www.paddlepaddle.org.cn/install/quick) if the PaddlePaddle on the device is lower than 2.0.0rc1.
Python 3.x, CUDA 10.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](http://www.paddlepaddle.org.cn/install/quick) if the PaddlePaddle on the device is lower than 2.0.0.
### Install PaddlePaddle using pip
...
...
@@ -16,13 +16,13 @@ Python 3.x, CUDA 10.0, CUDNN7.6.4 nccl2.1.2 and later version are required at fi
If you want to use PaddlePaddle on GPU, you can use the following command to install PaddlePaddle.
```bash
pip install paddlepaddle-gpu==2.0.0rc1--upgrade
pip install paddlepaddle-gpu==2.0.0 --upgrade
```
If you want to use PaddlePaddle on CPU, you can use the following command to install PaddlePaddle.
- Make sure the compiled version is later than PaddlePaddle2.0rc1.
- Make sure the compiled version is later than PaddlePaddle2.0.
- Indicate **WITH_DISTRIBUTE=ON** when compiling, Please refer to [Instruction](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/install/Tables.html#id3) 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.