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,依赖库环境版本要求如下:
2.1 CPU 版的 PaddlePaddle¶
python3 -m pip install paddlepaddle==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/
2.2 GPU 版的 PaddlePaddle¶
2.2.1 CUDA11.8 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
2.2.2 CUDA12.3 的 PaddlePaddle
python3 -m pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu123/
注:
飞桨对于主流各 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 install https://paddle-wheel.bj.bcebos.com/3.0.0-beta0/linux/linux-cpu-openblas-avx/paddlepaddle-3.0.0b1-cp38-cp38-linux_x86_64.whl
三、验证安装¶
安装完成后您可以使用 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