Update install.md

This commit is contained in:
littletomatodonkey 2021-01-05 19:58:17 +08:00 committed by GitHub
parent 4c44536ee3
commit d27a70df89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -9,12 +9,18 @@
## 二、安装PaddlePaddle
运行PaddleClas需要`PaddlePaddle 2.0rc`或更高版本。请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
运行PaddleClas需要`PaddlePaddle 2.0rc1`或更高版本。请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
如果已经安装好了cuda、cudnn、nccl或者安装好了docker、nvidia-docker运行环境可以pip安装最新GPU版本PaddlePaddle
```bash
pip install paddlepaddle-gpu --upgrade
pip install paddlepaddle-gpu==2.0.0rc1 --upgrade
```
如果希望在CPU环境中使用PaddlePaddle可以运行下面的命令安装PaddlePaddle。
```bash
pip install paddlepaddle==2.0.0rc1 --upgrade
```
也可以从源码编译安装PaddlePaddle请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
@ -33,13 +39,13 @@ python -c "import paddle; print(paddle.__version__)"
```
注意:
- 从源码编译的PaddlePaddle版本号为0.0.0请确保使用了PaddlePaddle 2.0rc及之后的源码编译。
- 从源码编译的PaddlePaddle版本号为0.0.0请确保使用了PaddlePaddle 2.0rc1及之后的源码编译。
- PaddleClas基于PaddlePaddle高性能的分布式训练能力若您从源码编译请确保打开编译选项**WITH_DISTRIBUTE=ON**。具体编译选项参考[编译选项表](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/install/Tables.html#id3)。
- 在docker中运行时为保证docker容器有足够的共享内存用于Paddle的数据读取加速在创建docker容器时请设置参数`--shm_size=8g`,条件允许的话可以设置为更大的值。
**运行环境需求:**
- Python3 (当前只支持Linux系统)
- Python3
- CUDA >= 9.0
- cuDNN >= 7.6.4
- nccl >= 2.1.2