2020-04-10 00:45:02 +08:00
|
|
|
|
# 安装说明
|
|
|
|
|
|
|
|
|
|
---
|
|
|
|
|
|
2020-04-21 22:32:17 +08:00
|
|
|
|
## 一、简介
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-04-21 22:32:17 +08:00
|
|
|
|
本章将介绍如何安装PaddleClas及其依赖项。
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
|
|
|
|
|
2020-04-21 22:32:17 +08:00
|
|
|
|
## 二、安装PaddlePaddle
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-11-10 20:36:10 +08:00
|
|
|
|
运行PaddleClas需要`PaddlePaddle 2.0rc`或更高版本。请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-04-22 22:58:35 +08:00
|
|
|
|
如果已经安装好了cuda、cudnn、nccl或者安装好了docker、nvidia-docker运行环境,可以pip安装最新GPU版本PaddlePaddle
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-04-15 18:54:00 +08:00
|
|
|
|
```bash
|
|
|
|
|
pip install paddlepaddle-gpu --upgrade
|
2020-04-10 00:45:02 +08:00
|
|
|
|
```
|
|
|
|
|
|
2020-04-22 22:58:35 +08:00
|
|
|
|
也可以从源码编译安装PaddlePaddle,请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-04-22 22:58:35 +08:00
|
|
|
|
使用以下命令可以验证PaddlePaddle是否安装成功。
|
2020-04-15 18:54:00 +08:00
|
|
|
|
|
|
|
|
|
```python
|
2020-11-10 20:36:10 +08:00
|
|
|
|
import paddle
|
|
|
|
|
paddle.utils.run_check()
|
2020-04-10 00:45:02 +08:00
|
|
|
|
```
|
2020-04-15 18:54:00 +08:00
|
|
|
|
|
2020-04-22 22:58:35 +08:00
|
|
|
|
查看PaddlePaddle版本的命令如下:
|
2020-04-15 18:54:00 +08:00
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
python -c "import paddle; print(paddle.__version__)"
|
|
|
|
|
```
|
|
|
|
|
|
2020-04-10 00:45:02 +08:00
|
|
|
|
注意:
|
2020-11-10 20:36:10 +08:00
|
|
|
|
- 从源码编译的PaddlePaddle版本号为0.0.0,请确保使用了PaddlePaddle 2.0rc及之后的源码编译。
|
2020-12-14 21:43:45 +08:00
|
|
|
|
- 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`,条件允许的话可以设置为更大的值。
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-04-15 18:54:00 +08:00
|
|
|
|
**运行环境需求:**
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
2020-06-04 20:08:13 +08:00
|
|
|
|
- Python3 (当前只支持Linux系统)
|
2020-04-15 18:54:00 +08:00
|
|
|
|
- CUDA >= 9.0
|
2020-12-14 21:43:45 +08:00
|
|
|
|
- cuDNN >= 7.6.4
|
2020-04-10 00:45:02 +08:00
|
|
|
|
- nccl >= 2.1.2
|
|
|
|
|
|
|
|
|
|
|
2020-04-21 22:32:17 +08:00
|
|
|
|
## 三、安装PaddleClas
|
2020-04-10 00:45:02 +08:00
|
|
|
|
|
|
|
|
|
**克隆PaddleClas模型库:**
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
cd path_to_clone_PaddleClas
|
|
|
|
|
git clone https://github.com/PaddlePaddle/PaddleClas.git
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
**安装Python依赖库:**
|
|
|
|
|
|
|
|
|
|
Python依赖库在[requirements.txt](https://github.com/PaddlePaddle/PaddleClas/blob/master/requirements.txt)中给出,可通过如下命令安装:
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
pip install --upgrade -r requirements.txt
|
|
|
|
|
```
|
2020-06-04 20:08:13 +08:00
|
|
|
|
|
|
|
|
|
visualdl可能出现安装失败,请尝试
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
pip3 install --upgrade visualdl==2.0.0b3 -i https://mirror.baidu.com/pypi/simple
|
|
|
|
|
|
|
|
|
|
```
|
2020-10-19 13:05:34 +08:00
|
|
|
|
|
|
|
|
|
此外,visualdl目前只支持在python3下运行,因此如果希望使用visualdl,需要使用python3。
|