Ubuntu安装PaddlePaddle 2.0 GPU版

目前，Linux发行版当中，Ubuntu18.04是最受欢迎的系统，所以本文介绍Ubuntu18.04下安装PaddlePaddle2.0 GPU版的步骤。

 

对于桌面版的Linux系统，最令人头疼的就是我的某个硬件内核里没有它的驱动。

虽然从16.04以后桌面版的Ubuntu能友好的支持大部分硬件了，但是像显卡这种东西，还是需要我们自己手动安装驱动的。

稍有不慎就会无限卡住在登陆界面，这也是曾经我在玩Ubuntu时候的噩梦，一言不合就重装系统，也成就了我如今安装各种系统都能十分熟练(熟练的令人心疼)

 

所以，Ubuntu安装PaddlePaddle的难点并不是在于Paddle本身，而是在安装显卡驱动、CUDA、cuDNN当中的各种坑。

 

我的环境
 

操作系统：Ubuntu 18.04.5 LTS  64位系统

（Paddle2.0支持

• Ubuntu 14.04 / 16.04 / 18.04

• CentOS 6 / 7

）[不推荐使用Ubuntu20.04，以为它自带的Python版本是3.9]

 

CPU: i7-8559U(支持avx指令集)

GPU: RTX2070

 

Python版本：3.6.9(Ubuntu 18.04.5内置版本)

注意Paddle目前对Python的要求:（强烈不建议使用Python2）

Python 3 的版本要求 3.5.1+/3.6/3.7/3.8

Python 具有 pip, 且 pip 的版本要求 20.2.2+

Python 和 pip 要求是 64 位版本

 

 

安装NVIDIA显卡驱动

 

假设你的Ubuntu是刚刚安装好的，并没有配置软件源，那么我们需要更换国内的软件源，不然下载速度会非常的慢。

 

首先打开软件与更新

 

在：“下载自”里找到其他站点，选择一个国内软件源(我使用的清华源)：

选择后会让你输入用户密码，然后点击关闭，按照提示更新一下。

这个步骤相当于：

sudo apt update

然后再次打开软件与更新，进入“附加驱动”：

 

如果你不知道选择那一个，可以直接在终端内输入命令：

sudo ubuntu-drivers autoinstall

让系统自动帮你选择版本安装。

 

选择驱动后就可以下载，等待安装成功后重新启动即可完成显卡驱动安装。

 

安装CUDA10.2

官方获取CUDA：https://developer.nvidia.com/cuda-toolkit-archive

可以不进入官方，直接通过如下命令下载并安装(官网搬运来的)：

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

如果下载速度慢可使用 https://pan.baidu.com/s/1P8IQ4ly8v6n1k2wEmvwzSg  提取码: m6j5

sudo sh cuda_10.2.89_440.33.01_linux.run

 如果出现下图提醒，直接忽视即可： 选择Continue

输入accept：

切记一定要取消Driver前面的[X] 

切记一定要取消Driver前面的[X] 

切记一定要取消Driver前面的[X] （如果你不想无限卡登陆界面循环的话）

这里只安装CUDA Toolkit 10.2 满足Paddle开发：然后选择Install

如果出现如下提示，恭喜你，CUDA安装成功了！

 

验证cuda安装成功：

nvcc -V

 

安装cuDNN

官方获取cuDNN:https://developer.nvidia.com/zh-cn/cudnn(下载需要注册NVIDIA账号)

方便读者下载已经上传至百度网盘： https://pan.baidu.com/s/1P8IQ4ly8v6n1k2wEmvwzSg  提取码: m6j5

 

首先解压：

tar -zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz 

解压后得到如下文件：

cuda
├── include
│   └── cudnn.h
├── lib64
│   ├── libcudnn.so -> libcudnn.so.7
│   ├── libcudnn.so.7 -> libcudnn.so.7.6.5
│   ├── libcudnn.so.7.6.5
│   └── libcudnn_static.a
└── NVIDIA_SLA_cuDNN_Support.txt

2 directories, 6 files

 

然后复制这些文件到CUDA10.2的目录下： 

sudo cp cuda/lib64/* /usr/local/cuda-10.2/lib64/

sudo cp cuda/include/* /usr/local/cuda-10.2/include/

 

最后验证cuDNN安装成功：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

安装PaddlePaddle GPU版

在终端内输入：

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

 

安装完成后在终端内输入 python3 进入python解释器

输入

import paddle

再输入 

paddle.utils.run_check()

 

PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now！

 

最后：

你问我为什么不讲CPU安装的？ 你都能自己安装一个Ubuntu的主机了，安装CPU版的Paddle还不是信手拈来的[手动狗头]、

 

安装CPU版Paddle 2.0

忽略以上所有步骤，然后在终端内输入：

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

 

 

---

官方Linux下pip安装PaddlePaddle文档：https://www.paddlepaddle.org.cn/documentation/docs/zh/install/pip/linux-pip.html