Linux 下的 PIP 安装¶
The Python Package Index(PyPI)是 Python 的包管理器。本文档为你介绍 PyPI 安装方式,飞桨提供的 PyPI 安装包支持分布式训练(多机多卡)、TensorRT 推理功能。
一、环境准备¶
1.1 如何查看您的环境¶
可以使用以下命令查看本机的操作系统和位数信息:
uname -m && cat /etc/*release
确认需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您计算机可能有多个 Python
根据您的环境您可能需要将说明中所有命令行中的 python3 替换为具体的 Python 路径
which python3
需要确认 python 的版本是否满足要求
使用以下命令确认是 3.8/3.9/3.10/3.11/3.12
python3 --version
需要确认 pip 的版本是否满足要求,要求 pip 版本为 20.2.2 或更高版本
python3 -m ensurepip
python3 -m pip --version
需要确认 Python 和 pip 是 64bit,并且处理器架构是 x86_64(或称作 x64、Intel 64、AMD64)架构。下面的第一行输出的是”64bit”,第二行输出的是”x86_64”、”x64”或”AMD64”即可:
python3 -c "import platform;print(platform.architecture()[0]);print(platform.machine())"
默认提供的安装包需要计算机支持 MKL
二、开始安装¶
首先请选择您的版本¶
如果您的计算机没有 NVIDIA® GPU,请安装CPU 版的 PaddlePaddle
如果您的计算机有 NVIDIA® GPU,请确保满足以下条件并且安装GPU 版 PaddlePaddle,依赖库环境版本要求如下:
CUDA 工具包 11.2 配合 cuDNN v8.2.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.0.3.4
CUDA 工具包 11.6 配合 cuDNN v8.4.0, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.0.6
CUDA 工具包 11.7 配合 cuDNN v8.4.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.2.4
CUDA 工具包 11.8 配合 cuDNN v8.6.0, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.5.1.7
CUDA 工具包 12.0 配合 cuDNN v8.9.1, 如需使用 PaddleTensorRT 推理,需配合 TensorRT8.6.1.6
如需使用分布式多卡环境,需配合 NCCL>=2.7
GPU 运算能力超过 6.0 的硬件设备
您可参考 NVIDIA 官方文档了解 CUDA、CUDNN 和 TensorRT 的安装流程和配置方法,请见CUDA,cuDNN,TensorRT
如果您需要使用多卡环境请确保您已经正确安装 nccl2,或者按照以下指令安装 nccl2(这里提供的是 CUDA11.2,cuDNN7 下 nccl2 的安装指令,更多版本的安装信息请参考 NVIDIA官方网站):
rm -f /usr/local/lib/libnccl.so wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-2.10.3-1+cuda11.4.x86_64.rpm wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-devel-2.10.3-1+cuda11.4.x86_64.rpm wget --no-check-certificate -q https://nccl2-deb.cdn.bcebos.com/libnccl-static-2.10.3-1+cuda11.4.x86_64.rpm rpm -ivh libnccl-2.10.3-1+cuda11.4.x86_64.rpm rpm -ivh libnccl-devel-2.10.3-1+cuda11.4.x86_64.rpm rpm -ivh libnccl-static-2.10.3-1+cuda11.4.x86_64.rpm
2.1 CPU 版的 PaddlePaddle¶
python3 -m pip install paddlepaddle==2.6.1 -i https://mirror.baidu.com/pypi/simple
2.2 GPU 版的 PaddlePaddle¶
2.2.1 CUDA11.2 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
CUDA11.2 包含 cuDNN 动态链接库的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post112 -f https://www.paddlepaddle.org.cn/whl/linux/cudnnin/stable.html
2.2.3 CUDA11.6 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post116 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
CUDA11.6 包含 cuDNN 动态链接库的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post116 -f https://www.paddlepaddle.org.cn/whl/linux/cudnnin/stable.html
2.2.4 CUDA11.7 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post117 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
CUDA11.7 包含 cuDNN 动态链接库的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post117 -f https://www.paddlepaddle.org.cn/whl/linux/cudnnin/stable.html
2.2.5 CUDA11.8 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1 -i https://mirror.baidu.com/pypi/simple
CUDA11.8 包含 cuDNN 动态链接库的 PaddlePaddle,需要先使用如下命令将 wheel 包下载到本地,再使用`python3 -m pip install [name].whl`本地安装([name]为 wheel 包名称):
python3 -m pip download paddlepaddle-gpu==2.6.1 -f https://www.paddlepaddle.org.cn/whl/linux/cudnnin/stable.html --no-index --no-deps
2.2.6 CUDA12.0 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
CUDA12.0 包含 cuDNN 动态链接库的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/linux/cudnnin/stable.html
注:
飞桨对于主流各 python 版本均提供了对应的安装包,而您环境中可能有多个 Python,请确认你想使用的 python 版本并下载对应的 paddlepaddle 安装包。例如您想使用 python3.10 的环境,则安装命令为 python3.10 -m pip install paddlepaddle。
上述命令默认安装
avx
、mkl
的包。判断你的机器是否支持avx
,可以输入以下命令,如果输出中包含avx
,则表示机器支持avx
。飞桨不再支持noavx
指令集的安装包。cat /proc/cpuinfo | grep -i avx
如果你想安装
avx
、openblas
的 Paddle 包,可以通过以下命令将 wheel 包下载到本地,再使用python3 -m pip install [name].whl
本地安装([name]为 wheel 包名称):python3 -m pip download paddlepaddle==2.6.1 -f https://www.paddlepaddle.org.cn/whl/linux/openblas/avx/stable.html --no-index --no-deps
三、验证安装¶
安装完成后您可以使用 python3
进入 python 解释器,输入import paddle
,再输入 paddle.utils.run_check()
如果出现PaddlePaddle is installed successfully!
,说明您已成功安装。
四、如何卸载¶
请使用以下命令卸载 PaddlePaddle:
CPU 版本的 PaddlePaddle:
python3 -m pip uninstall paddlepaddle
GPU 版本的 PaddlePaddle:
python3 -m pip uninstall paddlepaddle-gpu