文档简介:
The Python Package Index(PyPI)是 Python 的包管理器。本文档为你介绍 PyPI 安装方式,飞桨提供的 PyPI 安装包支持 TensorRT 推理功能。
一、环境准备
1.1 如何查看您的环境
-
需要确认 python 的版本是否满足要求
-
使用以下命令确认是 3.6/3.7/3.8/3.9/3.10
python --version
-
-
需要确认 pip 的版本是否满足要求,要求 pip 版本为 20.2.2 或更高版本
python -m ensurepip
python -m pip --version
-
需要确认 Python 和 pip 是 64bit,并且处理器架构是 x86_64(或称作 x64、Intel 64、AMD64)架构。下面的第一行输出的是”64bit”,第二行输出的是”x86_64”、”x64”或”AMD64”即可:
python -c "import platform;print(platform.architecture()[0]);print(platform.machine())"
-
默认提供的安装包需要计算机支持 MKL
-
Windows 暂不支持 NCCL,分布式等相关功能
二、开始安装
本文档为您介绍 pip 安装方式
首先请您选择您的版本
-
如果您的计算机没有 NVIDIA® GPU,请安装CPU 版的 PaddlePaddle
-
如果您的计算机有 NVIDIA® GPU,请确保满足以下条件并且安装 GPU 版 PaddlePaddle
-
CUDA 工具包 10.2 配合 cuDNN v7.6.5,如需使用 PaddleTensorRT 推理,需配合 TensorRT7.0.0.11
-
CUDA 工具包 11.2 配合 cuDNN v8.2.1,如需使用 PaddleTensorRT 推理,需配合 TensorRT8.2.4.2
-
CUDA 工具包 11.6 配合 cuDNN v8.4.0,如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.0.6
-
CUDA 工具包 11.7 配合 cuDNN v8.4.1,如需使用 PaddleTensorRT 推理,需配合 TensorRT8.4.2.4
-
GPU 运算能力超过 3.5 的硬件设备
-
注:目前官方发布的 windows 安装包仅包含 CUDA 10.2/11.2/11.6/11.7,如需使用其他 cuda 版本,请通过源码自行编译。您可参考 NVIDIA 官方文档了解 CUDA、CUDNN 和 TensorRT 的安装流程和配置方法,请见CUDA,cuDNN,TensorRT
-
根据版本进行安装
确定您的环境满足条件后可以开始安装了,选择下面您要安装的 PaddlePaddle
2.1 CPU 版的 PaddlePaddle
python -m pip install paddlepaddle==2.4.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
2.2 GPU 版的 PaddlePaddle
2.2.1 CUDA10.2 的 PaddlePaddle
python -m pip install paddlepaddle-gpu==2.4.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
2.2.2 CUDA11.2 的 PaddlePaddle
python -m pip install paddlepaddle-gpu==2.4.2.post112 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
2.2.3 CUDA11.6 的 PaddlePaddle
python -m pip install paddlepaddle-gpu==2.4.2.post116 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
2.2.4 CUDA11.7 的 PaddlePaddle
python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
注:
-
如果你使用的是安培架构的 GPU,推荐使用 CUDA11 以上。如果你使用的是非安培架构的 GPU,推荐使用 CUDA10.2,性能更优。
-
请确认需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您计算机可能有多个 Python。根据您的环境您可能需要将说明中所有命令行中的 python 替换为具体的 Python 路径。
-
上述命令默认安装avx的包。如果你的机器不支持avx,需要安装noavx的 Paddle 包。判断你的机器是否支持avx,可以安装CPU-Z工具查看“处理器-指令集”。
首先使用如下命令将 wheel 包下载到本地,仅支持 python3.8
-
cpu、mkl 版本 noavx 机器安装:
python -m pip download paddlepaddle==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows
-
cpu、openblas 版本 noavx 机器安装:
python -m pip download paddlepaddle==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/openblas
-
gpu 版本 cuda10.2 noavx 机器安装:
python -m pip download paddlepaddle-gpu==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/
再使用python -m pip install [name].whl本地安装([name]为 wheel 包名称)
-
-
noavx/stable.html --no-index --no-deps
-
/noavx/stable.html --no-index --no-deps
-
/mkl/noavx/stable.html --no-index --no-deps
-
如果你想安装avx、openblas的 Paddle 包,可以通过以下命令将 wheel 包下载到本地,再使用python -m pip install [name].whl本地安装([name]为 wheel 包名称):
python -m pip download paddlepaddle==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/openblas
-
/avx/stable.html --no-index --no-deps
三、验证安装
安装完成后您可以使用 python 进入 python 解释器,输入import paddle ,再输入 paddle.utils.run_check()
如果出现PaddlePaddle is installed successfully!,说明您已成功安装。
四、如何卸载
请使用以下命令卸载 PaddlePaddle:
-
CPU 版本的 PaddlePaddle: python -m pip uninstall paddlepaddle
-
GPU 版本的 PaddlePaddle: python -m pip uninstall paddlepaddle-gpu