install.md 2.4 KB
Newer Older
W
WuHaobo 已提交
1 2 3 4
# 安装说明

---

littletomatodonkey's avatar
littletomatodonkey 已提交
5
## 一、简介
W
WuHaobo 已提交
6

littletomatodonkey's avatar
littletomatodonkey 已提交
7
本章将介绍如何安装PaddleClas及其依赖项。
W
WuHaobo 已提交
8 9


littletomatodonkey's avatar
littletomatodonkey 已提交
10
## 二、安装PaddlePaddle
W
WuHaobo 已提交
11

L
littletomatodonkey 已提交
12
运行PaddleClas需要`PaddlePaddle 2.0`或更高版本。请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。
W
WuHaobo 已提交
13

L
littletomatodonkey 已提交
14 15
### 通过pip安装PaddlePaddle

D
dyning 已提交
16
如果已经安装好了cuda、cudnn、nccl或者安装好了docker、nvidia-docker运行环境,可以pip安装最新GPU版本PaddlePaddle
W
WuHaobo 已提交
17

W
wangshipeng01 已提交
18
```bash
L
littletomatodonkey 已提交
19
pip install paddlepaddle-gpu==2.0.0 --upgrade
L
littletomatodonkey 已提交
20 21 22 23 24
```

如果希望在CPU环境中使用PaddlePaddle,可以运行下面的命令安装PaddlePaddle。

```bash
L
littletomatodonkey 已提交
25
pip install paddlepaddle==2.0.0 --upgrade
W
WuHaobo 已提交
26 27
```

L
littletomatodonkey 已提交
28 29 30 31
### 源码编译PaddlePaddle

您也可以从源码编译安装PaddlePaddle,请参照[安装文档](http://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。

W
WuHaobo 已提交
32

D
dyning 已提交
33
使用以下命令可以验证PaddlePaddle是否安装成功。
W
wangshipeng01 已提交
34 35

```python
36 37
import paddle
paddle.utils.run_check()
W
WuHaobo 已提交
38
```
W
wangshipeng01 已提交
39

D
dyning 已提交
40
查看PaddlePaddle版本的命令如下:
W
wangshipeng01 已提交
41 42 43 44 45

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

W
WuHaobo 已提交
46
注意:
L
littletomatodonkey 已提交
47
- 从源码编译的PaddlePaddle版本号为0.0.0,请确保使用了PaddlePaddle 2.0及之后的源码编译。
48 49
- 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`,条件允许的话可以设置为更大的值。
W
WuHaobo 已提交
50

W
wangshipeng01 已提交
51
**运行环境需求:**
W
WuHaobo 已提交
52

L
littletomatodonkey 已提交
53
- Python3
W
wangshipeng01 已提交
54
- CUDA >= 9.0
55
- cuDNN >= 7.6.4
W
WuHaobo 已提交
56 57 58
- nccl >= 2.1.2


littletomatodonkey's avatar
littletomatodonkey 已提交
59
## 三、安装PaddleClas
W
WuHaobo 已提交
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74

**克隆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
```
S
fix  
shippingwang 已提交
75 76 77 78 79 80 81

visualdl可能出现安装失败,请尝试

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

```
82 83

此外,visualdl目前只支持在python3下运行,因此如果希望使用visualdl,需要使用python3。