|
73 | 73 |
|
74 | 74 | * 如果您的计算机没有 NVIDIA® GPU,请安装[CPU 版的 PaddlePaddle](#cpu) |
75 | 75 |
|
76 | | -* 如果您的计算机有 NVIDIA® GPU,请确保满足以下条件并且安装[GPU 版 PaddlePaddle](#gpu) |
| 76 | +* 如果您的计算机有 NVIDIA® GPU,请确保满足以下条件并且安装[GPU 版 PaddlePaddle](#gpu),依赖库环境版本要求如下: |
77 | 77 |
|
78 | | - * **CUDA 工具包 10.1 配合 cuDNN 7 (cuDNN 版本>=7.6.5, 如需多卡支持,需配合 NCCL2.7 及更高; 不支持使用 TensorRT)** |
| 78 | + * **CUDA 工具包 10.1 配合 cuDNN 7 (cuDNN 版本>=7.6.5), 不支持使用 TensorRT** |
79 | 79 |
|
80 | | - * **CUDA 工具包 10.2 配合 cuDNN 7 (cuDNN 版本>=7.6.5, 如需多卡支持,需配合 NCCL2.7 及更高;如需使用 PaddleTensorRT 推理,需配合 TensorRT7.0.0.11)** |
| 80 | + * **CUDA 工具包 10.2 配合 cuDNN 7 (cuDNN 版本>=7.6.5), 如需使用 PaddleTensorRT 推理,需配合 TensorRT7.0.0.11** |
81 | 81 |
|
82 | | - * **CUDA 工具包 11.1 配合 cuDNN v8.1.1(如需多卡支持,需配合 NCCL2.7 及更高;如需使用 PaddleTensorRT 推理,需配合 TensorRT7.2.3.4)** |
| 82 | + * **CUDA 工具包 11.1 配合 cuDNN v8.1.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT7.2.3.4** |
83 | 83 |
|
84 | | - * **CUDA 工具包 11.2 配合 cuDNN v8.1.1(如需多卡支持,需配合 NCCL2.7 及更高;如需使用 PaddleTensorRT 推理,需配合 TensorRT8.0.3.4)** |
| 84 | + * **CUDA 工具包 11.2 配合 cuDNN v8.1.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.0.3.4** |
85 | 85 |
|
86 | | - * **CUDA 工具包 11.6 配合 cuDNN v8.4.0(如需多卡支持,需配合 NCCL2.7 及更高;如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.0.6)** |
| 86 | + * **CUDA 工具包 11.6 配合 cuDNN v8.4.0, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.0.6** |
87 | 87 |
|
88 | | - * **CUDA 工具包 11.7 配合 cuDNN v8.4.1(如需多卡支持,需配合 NCCL2.7 及更高;如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.2.4)** |
| 88 | + * **CUDA 工具包 11.7 配合 cuDNN v8.4.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.2.4** |
| 89 | +
|
| 90 | + * **如需使用分布式多卡环境,需配合 NCCL>=2.7** |
89 | 91 |
|
90 | 92 | * **GPU 运算能力超过 3.5 的硬件设备** |
91 | 93 |
|
|
163 | 165 |
|
164 | 166 | 2.2.4 CUDA11.2 的 PaddlePaddle |
165 | 167 |
|
166 | | - cuDNN8.1.1: |
| 168 | + 如果您只进行训练,可使用cuDNN8.1.1 版本的飞桨: |
167 | 169 | ``` |
168 | 170 | python3 -m pip install paddlepaddle-gpu==2.4.0rc0.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html |
169 | 171 | ``` |
|
193 | 195 |
|
194 | 196 | * 如果你使用的是安培架构的 GPU,推荐使用 CUDA11 以上。如果你使用的是非安培架构的 GPU,推荐使用 CUDA10.2,性能更优。 |
195 | 197 |
|
196 | | -* 请确认需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您计算机可能有多个 Python。根据您的环境您可能需要将说明中所有命令行中的 python3 替换为具体的 Python 路径。 |
| 198 | +* 飞桨对于主流各python 版本均提供了对应的安装包,而您环境中可能有多个 Python,请确认你想使用的python 版本并下载对应的paddlepaddle 安装包。例如您想使用python3.7 的环境,则安装命令为python3.7 -m pip install paddlepaddle。 |
197 | 199 |
|
198 | 200 | * 如果您需要使用清华源,可以通过以下命令 |
199 | 201 |
|
200 | 202 | ``` |
201 | 203 | python3 -m pip install paddlepaddle-gpu==[版本号] -i https://pypi.tuna.tsinghua.edu.cn/simple |
202 | 204 | ``` |
203 | 205 |
|
204 | | -* 上述命令默认安装`avx`的包。如果你的机器不支持`avx`,需要安装`noavx`的 Paddle 包,可以通过以下命令安装,仅支持 python3.8: |
| 206 | +* 上述命令默认安装`avx`的包。如果你的机器不支持`avx`,需要安装`noavx`的 Paddle 包,判断你的机器是否支持`avx`,可以输入以下命令,如果输出中包含`avx`,则表示机器支持`avx` |
| 207 | + ``` |
| 208 | + cat /proc/cpuinfo | grep -i avx |
| 209 | + ``` |
205 | 210 |
|
206 | | - 首先使用如下命令将 wheel 包下载到本地,再使用`python3 -m pip install [name].whl`本地安装([name]为 wheel 包名称): |
| 211 | + 首先使用如下命令将 wheel 包下载到本地: |
207 | 212 |
|
208 | 213 | * cpu、mkl 版本 noavx 机器安装: |
209 | 214 |
|
|
229 | 234 | ``` |
230 | 235 | python3 -m pip download paddlepaddle-gpu==2.4.0rc0 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/noavx/stable.html --no-index --no-deps |
231 | 236 | ``` |
| 237 | + |
| 238 | + 再使用`python3 -m pip install [name].whl`本地安装([name]为 wheel 包名称)。 |
232 | 239 |
|
233 | | - 判断你的机器是否支持`avx`,可以输入以下命令,如果输出中包含`avx`,则表示机器支持`avx` |
234 | | - ``` |
235 | | - cat /proc/cpuinfo | grep -i avx |
236 | | - ``` |
237 | 240 |
|
238 | 241 | * 如果你想安装`avx`、`openblas`的 Paddle 包,可以通过以下命令将 wheel 包下载到本地,再使用`python3 -m pip install [name].whl`本地安装([name]为 wheel 包名称): |
239 | 242 |
|
|
242 | 245 | ``` |
243 | 246 |
|
244 | 247 |
|
245 | | -
|
246 | | -
|
247 | 248 | ## **三、验证安装** |
248 | 249 |
|
249 | 250 | 安装完成后您可以使用 `python3` 进入 python 解释器,输入`import paddle` ,再输入 |
|
0 commit comments