Windows 下的 PIP 安装

The Python Package Index(PyPI)是 Python 的包管理器。本文档为你介绍 PyPI 安装方式，飞桨提供的 PyPI 安装包支持 TensorRT 推理功能。

一、环境准备

1.1 如何查看您的环境

  • 需要确认 python 的版本是否满足要求

    • 使用以下命令确认是 3.6/3.7/3.8/3.9/3.10

      1
      python --version

  • 需要确认 pip 的版本是否满足要求，要求 pip 版本为 20.2.2 或更高版本

    1
    python -m ensurepip
    1
    python -m pip --version

  • 需要确认 Python 和 pip 是 64bit，并且处理器架构是 x86_64（或称作 x64、Intel 64、AMD64）架构。下面的第一行输出的是”64bit”，第二行输出的是”x86_64”、”x64”或”AMD64”即可：

    1
    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

1
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

1
python -m pip install paddlepaddle-gpu==2.4.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

2.2.2 CUDA11.2 的 PaddlePaddle

1
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

1
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

1
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 机器安装：
    1
    python -m pip download paddlepaddle==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/noavx/stable.html --no-index --no-deps
    • cpu、openblas 版本 noavx 机器安装：
    1
    python -m pip download paddlepaddle==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/openblas/noavx/stable.html --no-index --no-deps
    • gpu 版本 cuda10.2 noavx 机器安装：
    1
    python -m pip download paddlepaddle-gpu==2.4.2 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/noavx/stable.html --no-index --no-deps

    再使用python -m pip install [name].whl本地安装（[name]为 wheel 包名称）

  • 如果你想安装avx、openblas的 Paddle 包，可以通过以下命令将 wheel 包下载到本地，再使用python -m pip install [name].whl本地安装（[name]为 wheel 包名称）：

    1
    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

第二种安装方式

Windows 下的 Conda 安装

Anaconda是一个免费开源的 Python 和 R 语言的发行版本，用于计算科学，Anaconda 致力于简化包管理和部署。Anaconda 的包使用软件包管理系统 Conda 进行管理。Conda 是一个开源包管理系统和环境管理系统，可在 Windows、macOS 和 Linux 上运行。本文档为你介绍 Anaconda 安装方式，飞桨提供的 Anaconda 安装包支持 TensorRT 推理功能。

一、环境准备

1.1 创建虚拟环境

1.1.1 安装环境

首先根据具体的 Python 版本创建 Anaconda 虚拟环境，PaddlePaddle 的 Anaconda 安装支持 3.6 - 3.10 版本的 Python 安装环境。

1
conda create -n paddle_env python=YOUR_PY_VER

1.1.2 进入 Anaconda 虚拟环境

1
activate paddle_env

1.2 其他环境检查

1.2.1 确认 Python 安装路径

确认您的 conda 虚拟环境和需要安装 PaddlePaddle 的 Python 是您预期的位置，因为您计算机可能有多个 Python。进入 Anaconda 的命令行终端，输入以下指令确认 Python 位置。

输出 Python 路径的命令为:

1
where python

根据您的环境，您可能需要将说明中所有命令行中的 python 替换为具体的 Python 路径

1.2.2 检查 Python 版本

使用以下命令确认版本

1
python --version

1.2.3 检查系统环境

确认 Python 和 pip 是 64bit，并且处理器架构是 x86_64（或称作 x64、Intel 64、AMD64）架构。下面的第一行输出的是”64bit”，第二行输出的是”x86_64（或 x64、AMD64）”即可：

1
python -c "import platform;print(platform.architecture()[0]);print(platform.machine())"

二、开始安装

本文档为您介绍 conda 安装方式

添加清华源（可选）

对于国内用户无法连接到 Anaconda 官方源的可以按照以下命令添加清华源:

1
2
3
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

根据版本进行安装

选择下面您要安装的 PaddlePaddle

CPU 版的 PaddlePaddle

如果您的计算机没有 NVIDIA® GPU，请安装 CPU 版的 PaddlePaddle

1
conda install paddlepaddle==2.4.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

GPU 版的 PaddlePaddle

  • 对于 CUDA 10.2，需要搭配 cuDNN 7.6.5，安装命令为:
1
conda install paddlepaddle-gpu==2.4.2 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/
  • 对于 CUDA 11.2，需要搭配 cuDNN 8.2.1，安装命令为:
1
conda install paddlepaddle-gpu==2.4.2 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge
  • 对于 CUDA 11.6，需要搭配 cuDNN 8.4.0，安装命令为:
1
conda install paddlepaddle-gpu==2.4.2 cudatoolkit=11.6 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge
  • 对于 CUDA 11.7，需要搭配 cuDNN 8.4.1，安装命令为:
1
conda install paddlepaddle-gpu==2.4.2 cudatoolkit=11.7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge

您可参考 NVIDIA 官方文档了解 CUDA 和 CUDNN 的安装流程和配置方法，请见CUDA，cuDNN

三、验证安装

安装完成后您可以使用 python 或 python3 进入 python 解释器，输入import paddle ，再输入 paddle.utils.run_check()

如果出现PaddlePaddle is installed successfully!，说明您已成功安装。