zl程序教程

您现在的位置是:首页 >  系统

当前栏目

Linux下安装百度Paddle

2023-09-14 09:09:13 时间

一、环境准备

1.1当前飞桨支持的环境

Linux 版本(64 位)

CentOS 7 (GPU 版本支持 CUDA 10.1/10.2/11.0/11.2)

Ubuntu 16.04(GPU版本支持CUDA 10.1/10.2/11.0/11.2)

Ubuntu 18.04(GPU版本支持CUDA 10.1/10.2/11.0/11.2)

Python 版本 3.6/3.7/3.8/3.9(64 位)

pip 或 pip3 版本 20.2.2 或更高版本(64 位)

1.2 如何查看您的环境

可以使用以下命令查看本机的信息和信息:

uname -m && cat /etc/*release
确认需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您的计算机可能有 Python

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

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

使用以下命令确认为 3.6/3.7/3.8/3.9

python --version
需要确认pip的版本是否满足要求,要求pip版本为20.2.2或更高版本

python -m ensurepip
python -m pip --version
需要确认 Python 和 64bit 处理器架构是 x86_64(或任何 x64、Intel 64、AMD64)架构,现在 PaddlePaddle 不支持 arm64 架构。下面的第一行输出是“64bit”,第二行输出是“x86_64”、“x64”或“AMD64”分类:

python -c “import platform;print(platform.architecture()[0]);print(platform.machine())”
默认提供的安装包需要计算机支持MKL

如果您对机器环境不了解,请下载使用快速安装脚本,配套说明请参考这里。

二、开始安装

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

首先请选择您的版本
如果您的计算机没有 NVIDIA® GPU,请安装CPU 版的 PaddlePaddle

如果您的计算机有 NVIDIA® GPU,请确保满足以下条件并且安装GPU 版 PaddlePaddle

CUDA工具包10.1/10.2配合cuDNN 7(cuDNN版本>=7.6.5,多卡支持,需配合NCCL2.7及以上)

CUDA工具包11.0配合cuDNN v8.0.4(如需多卡支持,需配合NCCL2.7及升级)

CUDA工具包11.2配合cuDNN v8.1.1(如需多卡支持,需配合NCCL2.7及升级)

GPU运算超过能力1.0的硬件设备

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

如果您需要使用多卡环境,请确保您正确安装nccl2,或者按照以下指令安装nccl2(这里提供的是CUDA10.2,cuDNN7下nccl2的安装指令,更多版本的安装信息请参考NVIDIA官方网站):

Centos系统可以参考以下命令

wget http://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm
rpm -i nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm
yum update -y
yum install -y libnccl-2.7.8-1+cuda10.2 libnccl-devel-2.7.8-1+cuda10.2 libnccl-static-2.7.8-1+cuda10.2
Ubuntu系统可以参考以下命令

wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
sudo apt install -y libnccl2=2.7.8-1+cuda10.2 libnccl-dev=2.7.8-1+cuda10.2

2.1 CPU版的PaddlePaddle

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

2.2 GPU版的PaddlePaddle

2.2.1 CUDA10.1 的 PaddlePaddle

python -m pip install paddlepaddle-gpu==2.1.2.post101 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

2.2.2 CUDA10.2的PaddlePaddle

python -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple

2.2.3 CUDA11.0 的 PaddlePaddle

python -m pip install paddlepaddle-gpu==2.1.2.post110 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

2.2.4 CUDA11.2的PaddlePaddle

python -m pip install paddlepaddle-gpu==2.1.2.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
注:

如果你使用的是安培架构的GPU,推荐使用CUDA11.2。如果你使用的是非安培架构的GPU,推荐使用CUDA10.2,性能更优。

请确认需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您的计算机可能有多个 Python。根据您的环境,您可能需要说明中所有命令行中的 Python 替换为 python3 或者替换为具体的 Python 路径。

如果您需要使用清华源,可以通过以下命令

python -m pip install paddlepaddle-gpu==[版本号] -i https://pypi.tuna.tsinghua.edu.cn/simple
命令上述默认安装avx的包如果你的机器不请立即获取iTunes。 avx,安装需要noavx的桨包,可以通过以下命令安装,仅支持python3.8:

首先使用如下命令将wheel包下载到本地,再使用本地安装([name]为wheel包名称):python -m pip install [name].whl

cpu、mkl版本noavx机器安装:

python -m pip download paddlepaddle==2.1.2 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/noavx/stable.html --no-index --no-deps
cpu、openblas版本noavx机器安装:

python -m pip download paddlepaddle==2.1.2 -f https://www.paddlepaddle.org.cn/whl/linux/openblas/noavx/stable.html --no-index --no-deps
gpu版本cuda10.1 noavx机器安装:

python -m pip download paddlepaddle-gpu==2.1.2.post101 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/noavx/stable.html --no-index --no-deps
gpu版本cuda10.2 noavx机器安装:

python -m pip download paddlepaddle-gpu==2.1.2 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/noavx/stable.html --no-index --no-deps
判断你的机器是否支持avx,可以输入以下命令,如果输出中包含avx,则表示机器支持avx

cat /proc/cpuinfo | grep -i avx
如果你想安装联编tensorrt的Paddle包,可以通过以下命令

python -m pip install paddlepaddle-gpu==[版本号] -f https://www.paddlepaddle.org.cn/whl/stable/tensorrt.html
如果你想安装avx、openblas的Paddle包,可以通过以下命令

python -m pip install paddlepaddle-gpu==[版本号] -f https://www.paddlepaddle.org.cn/whl/linux/openblas/avx/stable.html

三、验证安装

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

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

四、如何卸

请使用以下命令卸下PaddlePaddle:

CPU 版本的 PaddlePaddle :python -m pip uninstall paddlepaddle

GPU 版本的 PaddlePaddle :python -m pip uninstall paddlepaddle-gpu

参考:https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html